Home
last modified time | relevance | path

Searched refs:replace_by_example (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Ddecompose_mem_bound_mm.py107 match.replace_by_example(repl, [mat1, mat2])
130 match.replace_by_example(repl, [mat1, mat2, mat3])
150 match.replace_by_example(repl, [mat1, mat2])
H A Dreplace_random.py125 match.replace_by_example(replacement, [size])
145 match.replace_by_example(replacement, [low, high, size])
H A Dpost_grad.py458 match.replace_by_example(repl, list(shape))
830 match.replace_by_example(decomp, flat_args, run_functional_passes=False)
850 match.replace_by_example(decomp, flat_args, run_functional_passes=False)
874 match.replace_by_example(decomp, flat_args, run_functional_passes=False)
1002 match.replace_by_example(repl, [inp, mat1, mat2])
1045 match.replace_by_example(repl, [inp, mat1, mat2])
H A Djoint_graph.py659 match.replace_by_example(repl, [inp, other])
686 match.replace_by_example(repl, [inp, other])
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_pattern_matcher.py1229 match.replace_by_example(repl, [inp, mat1, mat2])
1279 match.replace_by_example(repl, [inp, mat1, mat2])
1327 match.replace_by_example(repl, [inp, mat1, mat2])
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dcomms.py437 match.replace_by_example(
H A Dpattern_matcher.py225 def replace_by_example( member in Match