Home
last modified time | relevance | path

Searched defs:add_Tensor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/serialized_patterns/
H A D_sfdp_pattern_16.py47 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.py41 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.py43 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.py46 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.py45 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