Searched refs:normalize_op_input_output2 (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_ops.py | 130 def normalize_op_input_output2( function
192 return normalize_op_input_output2(
263 fn, primals = normalize_op_input_output2(f, args, kwargs)
312 fn, primals = normalize_op_input_output2(
689 contig_fn, primals = normalize_op_input_output2(
717 noncontig_fn, primals = normalize_op_input_output2(
1722 fn, primals = normalize_op_input_output2(
2656 inner_vmapped_fn, primals = normalize_op_input_output2(
2662 inner_mapped_fn, _ = normalize_op_input_output2(
2714 vmapped_fn, primals = normalize_op_input_output2(
[all …]
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_decomp.py | 310 def normalize_op_input_output2( function
361 return normalize_op_input_output2(
|