Home
last modified time | relevance | path

Searched refs:MultiOutputPattern (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/serialized_patterns/
H A D_sfdp_pattern_18.py26 MultiOutputPattern,
93 _sfdp_pattern_18_training = MultiOutputPattern([view_default_5,
132 _sfdp_pattern_18_inference = MultiOutputPattern([view_default_5,
195 _sfdp_pattern_18_bs1_training = MultiOutputPattern([view_default_5,
231 _sfdp_pattern_18_bs1_inference = MultiOutputPattern([view_default_5,
302 _sfdp_pattern_18_half_training = MultiOutputPattern([view_default_5,
343 _sfdp_pattern_18_half_inference = MultiOutputPattern([view_default_5,
411 _sfdp_pattern_18_half_bs1_training = MultiOutputPattern([view_default_5,
449 _sfdp_pattern_18_half_bs1_inference = MultiOutputPattern([view_default_5,
H A Dbmm_pattern.py26 MultiOutputPattern,
38 bmm_pattern_training = MultiOutputPattern([bmm_default,
H A Dmm_pattern.py26 MultiOutputPattern,
38 mm_pattern_training = MultiOutputPattern([mm_default,
H A Daddmm_pattern.py26 MultiOutputPattern,
43 addmm_pattern_training = MultiOutputPattern([addmm_default,
H A D_sfdp_pattern_13.py26 MultiOutputPattern,
61 _sfdp_pattern_13_training = MultiOutputPattern([bmm_default_1,
112 _sfdp_pattern_13_half_training = MultiOutputPattern([bmm_default_1,
H A D_sfdp_pattern_1.py26 MultiOutputPattern,
72 _sfdp_pattern_1_training = MultiOutputPattern([view_default_5,
145 _sfdp_pattern_1_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_5.py26 MultiOutputPattern,
73 _sfdp_pattern_5_training = MultiOutputPattern([view_default_5,
148 _sfdp_pattern_5_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_2.py26 MultiOutputPattern,
72 _sfdp_pattern_2_training = MultiOutputPattern([view_default_5,
145 _sfdp_pattern_2_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_3.py26 MultiOutputPattern,
79 _sfdp_pattern_3_training = MultiOutputPattern([view_default_5,
160 _sfdp_pattern_3_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_4.py26 MultiOutputPattern,
79 _sfdp_pattern_4_training = MultiOutputPattern([view_default_5,
160 _sfdp_pattern_4_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_11.py26 MultiOutputPattern,
81 _sfdp_pattern_11_training = MultiOutputPattern([view_default_5,
169 _sfdp_pattern_11_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_6.py26 MultiOutputPattern,
80 _sfdp_pattern_6_training = MultiOutputPattern([view_default_5,
163 _sfdp_pattern_6_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_16.py26 MultiOutputPattern,
89 _sfdp_pattern_16_training = MultiOutputPattern([view_default_5,
180 _sfdp_pattern_16_bs1_training = MultiOutputPattern([view_default_5,
276 _sfdp_pattern_16_half_training = MultiOutputPattern([view_default_5,
374 _sfdp_pattern_16_half_bs1_training = MultiOutputPattern([view_default_5,
470 _sfdp_pattern_16_half_mask_fp32_training = MultiOutputPattern([view_default_5,
565 _sfdp_pattern_16_half_mask_fp32_bs1_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_7.py26 MultiOutputPattern,
91 _sfdp_pattern_7_training = MultiOutputPattern([view_default_5,
186 _sfdp_pattern_7_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_19.py26 MultiOutputPattern,
85 _sfdp_pattern_19_training = MultiOutputPattern([view_default_5,
175 _sfdp_pattern_19_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_10.py26 MultiOutputPattern,
84 _sfdp_pattern_10_training = MultiOutputPattern([view_default_5,
171 _sfdp_pattern_10_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_8.py26 MultiOutputPattern,
84 _sfdp_pattern_8_training = MultiOutputPattern([view_default_5,
171 _sfdp_pattern_8_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_14.py26 MultiOutputPattern,
82 _sfdp_pattern_14_training = MultiOutputPattern([view_default_5,
173 _sfdp_pattern_14_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_9.py26 MultiOutputPattern,
91 _sfdp_pattern_9_training = MultiOutputPattern([view_default_5,
186 _sfdp_pattern_9_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_12.py26 MultiOutputPattern,
88 _sfdp_pattern_12_training = MultiOutputPattern([view_default_5,
184 _sfdp_pattern_12_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_15.py26 MultiOutputPattern,
87 _sfdp_pattern_15_training = MultiOutputPattern([view_default_5,
187 _sfdp_pattern_15_half_training = MultiOutputPattern([view_default_5,
H A D_sfdp_pattern_17.py26 MultiOutputPattern,
94 _sfdp_pattern_17_training = MultiOutputPattern([view_default_5,
202 _sfdp_pattern_17_half_training = MultiOutputPattern([view_default_5,
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dpattern_matcher.py802 class MultiOutputPattern(PatternExpr): class
1816 return MultiOutputPattern(pytree.tree_leaves(pattern))