Searched refs:pass_patterns (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | freezing_patterns.py | 27 pass_patterns = [ variable
67 for pattern in pass_patterns:
103 pass_dict=pass_patterns[pass_number],
163 pass_patterns[0],
181 pass_patterns[0],
203 pass_patterns[0],
219 pass_dict=pass_patterns[0],
|
| D | post_grad.py | 65 pass_patterns = [ variable
99 for patterns in pass_patterns:
197 pattern, extra_check, pass_dict=pass_patterns[pass_number]
439 pass_dict=pass_patterns[1],
995 pass_dict=pass_patterns[2],
1029 pass_dict=pass_patterns[2],
1038 pass_dict=pass_patterns[2],
|
| D | joint_graph.py | 37 pass_patterns = [ variable
459 for patterns in pass_patterns:
665 pass_dict=pass_patterns[1],
692 pass_dict=pass_patterns[1],
|
| D | misc_patterns.py | 18 from .post_grad import pass_patterns as post_grad_patterns_all
|
| /external/pytorch/test/inductor/ |
| D | test_pattern_matcher.py | 1233 torch._inductor.fx_passes.post_grad.pass_patterns + [test_pass],
1283 torch._inductor.fx_passes.post_grad.pass_patterns + [test_pass],
1331 torch._inductor.fx_passes.post_grad.pass_patterns + [test_pass],
|