Searched defs:gradcheck_wrapper (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | composite_compliance.py | 416 def compute_expected_grads(op, args, kwargs, output_process_fn_grad=None, gradcheck_wrapper=None): 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
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/ |
| H A D | core.py | 804 gradcheck_wrapper: Callable = lambda op, *args, **kwargs: op(*args, **kwargs) variable in OpInfo
|