/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/serialized_patterns/ |
H A D | _sfdp_pattern_16.py | 47 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask'), _users=2) variable 111 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask'), _users=2) variable 139 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask'), _users=2) variable 200 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask'), _users=2) variable 229 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask')) variable 298 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask')) variable 328 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask')) variable 394 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask')) variable 425 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask'), _users=2) variable 492 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask'), _users=2) variable [all …]
|
H A D | _sfdp_pattern_5.py | 41 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask'), _users=2) variable 89 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask'), _users=2) variable 111 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask')) variable 164 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask')) variable
|
H A D | _sfdp_pattern_6.py | 43 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask'), _users=2) variable 97 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask'), _users=2) variable 121 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask')) variable 180 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask')) variable
|
H A D | _sfdp_pattern_19.py | 46 add_Tensor = CallFunction(aten.add.Tensor, where_self, KeywordArg('attn_mask'), _users=2) variable 106 add_Tensor = CallFunction(aten.add.Tensor, where_self, KeywordArg('attn_mask'), _users=2) variable 133 add_Tensor = CallFunction(aten.add.Tensor, where_self, KeywordArg('attn_mask'), _users=2) variable 196 add_Tensor = CallFunction(aten.add.Tensor, where_self, KeywordArg('attn_mask'), _users=2) variable
|
H A D | _sfdp_pattern_14.py | 45 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask'), _users=2) variable 103 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask'), _users=2) variable 131 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask')) variable 194 add_Tensor = CallFunction(aten.add.Tensor, div_Tensor, KeywordArg('attn_mask')) variable
|