Searched refs:construct_pattern_matcher_pass (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | split_cat.py | 89 def construct_pattern_matcher_pass(pass_name: str): function
206 pass_dict=construct_pattern_matcher_pass("normalization_pass"),
210 pass_dict=construct_pattern_matcher_pass("normalization_pass"),
218 pass_dict=construct_pattern_matcher_pass("remove_split_with_size_one_pass"),
222 pass_dict=construct_pattern_matcher_pass("remove_split_with_size_one_pass"),
256 pass_dict=construct_pattern_matcher_pass("normalization_pass"),
260 pass_dict=construct_pattern_matcher_pass("normalization_pass"),
296 pass_dict=construct_pattern_matcher_pass("normalization_pass"),
357 pass_dict=construct_pattern_matcher_pass("normalization_pass"),
402 pass_dict=construct_pattern_matcher_pass("normalization_pass"),
[all …]
|
| H A D | decompose_mem_bound_mm.py | 11 from .split_cat import construct_pattern_matcher_pass
97 pass_dict=construct_pattern_matcher_pass("decompose_mm_pass"),
115 pass_dict=construct_pattern_matcher_pass("decompose_mm_pass"),
138 pass_dict=construct_pattern_matcher_pass("decompose_mm_pass"),
|