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(
|