Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/serialized_patterns/
H A D_sfdp_pattern_18.py49 where_self = CallFunction(aten.where.self, KeywordArg('causal_mask'), div_Tensor, full_default_1, _… variable
118 where_self = CallFunction(aten.where.self, KeywordArg('causal_mask'), div_Tensor, full_default_1, _… variable
152 where_self = CallFunction(aten.where.self, KeywordArg('causal_mask'), div_Tensor, full_default_1, _… variable
218 where_self = CallFunction(aten.where.self, KeywordArg('causal_mask'), div_Tensor, full_default_1, _… variable
253 where_self = CallFunction(aten.where.self, KeywordArg('causal_mask'), div_Tensor, full_default_1) variable
327 where_self = CallFunction(aten.where.self, KeywordArg('causal_mask'), div_Tensor, full_default_1) variable
363 where_self = CallFunction(aten.where.self, KeywordArg('causal_mask'), div_Tensor, full_default_1) variable
434 where_self = CallFunction(aten.where.self, KeywordArg('causal_mask'), div_Tensor, full_default_1) variable
H A D_sfdp_pattern_19.py45 where_self = CallFunction(aten.where.self, KeywordArg('causal_mask'), div_Tensor, full_default_1) variable
105 where_self = CallFunction(aten.where.self, KeywordArg('causal_mask'), div_Tensor, full_default_1) variable
132 where_self = CallFunction(aten.where.self, KeywordArg('causal_mask'), div_Tensor, full_default_1) variable
195 where_self = CallFunction(aten.where.self, KeywordArg('causal_mask'), div_Tensor, full_default_1) variable
H A D_sfdp_pattern_15.py48 where_self = CallFunction(aten.where.self, expand_default, full_default, div_Tensor, _users=2) variable
112 where_self = CallFunction(aten.where.self, expand_default, full_default, div_Tensor, _users=2) variable
143 where_self = CallFunction(aten.where.self, expand_default, full_default, div_Tensor) variable
212 where_self = CallFunction(aten.where.self, expand_default, full_default, div_Tensor) variable
H A D_sfdp_pattern_17.py50 where_self = CallFunction(aten.where.self, expand_default, full_default, div_Tensor, _users=2) variable
120 where_self = CallFunction(aten.where.self, expand_default, full_default, div_Tensor, _users=2) variable
153 where_self = CallFunction(aten.where.self, expand_default, full_default, div_Tensor) variable
228 where_self = CallFunction(aten.where.self, expand_default, full_default, div_Tensor) variable
/aosp_15_r20/external/pytorch/torch/nested/_internal/
H A Dops.py974 def where_self(func, *args, **kwargs): function