Searched refs:PatternMatcherPass (Results 1 – 12 of 12) sorted by relevance
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | freezing_patterns.py | 17 PatternMatcherPass,
28 PatternMatcherPass(),
29 PatternMatcherPass(),
30 PatternMatcherPass(),
33 binary_folding_pass = PatternMatcherPass()
|
| D | pre_grad.py | 24 PatternMatcherPass,
35 efficient_conv_bn_eval_pass = PatternMatcherPass(
39 fuse_split_linear_add_pass = PatternMatcherPass(
42 fuse_chunk_squeeze_cat_pass = PatternMatcherPass(
45 remove_reshape_pass = PatternMatcherPass(
50 normalization_pass_aten = PatternMatcherPass()
51 merge_splits_pass_aten = PatternMatcherPass()
52 split_cat_pass_aten = PatternMatcherPass()
53 unbind_stack_pass_aten = PatternMatcherPass()
54 merge_getitem_cat_pass_aten = PatternMatcherPass()
[all …]
|
| D | replace_random.py | 13 PatternMatcherPass,
20 patterns = PatternMatcherPass()
|
| D | post_grad.py | 41 PatternMatcherPass,
66 PatternMatcherPass(),
67 PatternMatcherPass(),
68 PatternMatcherPass(),
808 graph_pass = PatternMatcherPass()
|
| D | joint_graph.py | 25 PatternMatcherPass,
33 patterns = PatternMatcherPass()
39 PatternMatcherPass(),
|
| D | b2b_gemm.py | 24 PatternMatcherPass,
36 B2B_GEMM_PASS = PatternMatcherPass(
|
| D | split_cat.py | 25 PatternMatcherPass,
44 PRE_GRAD_PATTERNS: Dict[str, PatternMatcherPass] = {}
45 POST_GRAD_PATTERNS: Dict[str, PatternMatcherPass] = {}
75 PRE_GRAD_PATTERNS[pass_name] = PatternMatcherPass(
84 POST_GRAD_PATTERNS[pass_name] = PatternMatcherPass(
|
| D | micro_pipeline_tp.py | 18 PatternMatcherPass,
23 patterns = PatternMatcherPass()
|
| /external/pytorch/test/inductor/ |
| D | test_custom_post_grad_passes.py | 12 from torch._inductor.pattern_matcher import Arg, CallFunction, PatternMatcherPass
110 class _CustomPass(PatternMatcherPass):
|
| D | test_pattern_matcher.py | 22 PatternMatcherPass,
964 PatternMatcherPass,
970 class _CustomPass(PatternMatcherPass):
1034 test_pass = PatternMatcherPass(pass_name="test")
1192 test_pass = PatternMatcherPass()
1249 test_pass = PatternMatcherPass()
1295 test_pass = PatternMatcherPass()
|
| /external/pytorch/torch/_inductor/ |
| D | comms.py | 362 PatternMatcherPass,
393 graph_pass = PatternMatcherPass()
|
| D | pattern_matcher.py | 990 elif isinstance(pass_dicts, (dict, PatternMatcherPass)):
1660 class PatternMatcherPass: class
1876 matcher_pass = PatternMatcherPass()
|