Searched defs:output_process_fn_grad (Results 1 – 5 of 5) 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
445 output_process_fn_grad=None, argument
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_ops.py | 131 f, args, kwargs, output_process_fn_grad=None, requires_grad=True argument
161 f, args, kwargs, sample_args, output_process_fn_grad=None argument
310 f, args, kwargs, output_process_fn_grad=None, requires_grad=False argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/ |
| H A D | core.py | 143 output_process_fn_grad=None, argument
195 self, *, output_process_fn_grad=None, broadcasts_input=None, name=None argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_decomp.py | 311 f, args, kwargs, output_process_fn_grad=None, requires_grad=True argument
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor.py | 384 output_process_fn_grad=lambda x: x, argument
588 output_process_fn_grad=lambda x: x, argument
|