Searched defs:assert_equal_fn (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | composite_compliance.py | 336 def check_all_permutations(op, args, kwargs, assert_equal_fn): argument
381 def check_with_mode(op, args, kwargs, assert_equal_fn): argument
446 gradcheck_wrapper=None, assert_equal_fn=None): argument
504 def check_forward_ad_formula(op: Callable, args, kwargs, gradcheck_wrapper=None, assert_equal_fn=No… argument
|