Home
last modified time | relevance | path

Searched refs:normalize_op_input_output2 (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_ops.py130 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 Dtest_decomp.py310 def normalize_op_input_output2( function
361 return normalize_op_input_output2(