Searched refs:normalize_op_input_output (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_ops.py | 190 def normalize_op_input_output(f, sample, requires_grad=True): function
225 fn, primals = normalize_op_input_output(f, sample)
258 fn, primals = normalize_op_input_output(f, sample)
291 fn, primals = normalize_op_input_output(f, sample, requires_grad=False)
322 fn, primals = normalize_op_input_output(f, sample, requires_grad=False)
806 fn, primals = normalize_op_input_output(_op, sample)
820 noncontig_fn, noncontig_primals = normalize_op_input_output(
1860 fn, primals = normalize_op_input_output(op, sample)
2107 fn, primals = normalize_op_input_output(op, sample)
2467 fn, primals = normalize_op_input_output(op, sample_input)
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_decomp.py | 359 def normalize_op_input_output(f, sample, requires_grad=True): function
967 fn, primals = normalize_op_input_output(func, sample_input)
|