Home
last modified time | relevance | path

Searched full:dim_intlist (Results 1 – 25 of 89) sorted by relevance

1234

/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/serialized_patterns/
H A D_sfdp_pattern_16.py51 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
72 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
115 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
143 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
163 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
204 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
234 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
258 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
303 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
333 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
[all …]
H A D_sfdp_pattern_18.py53 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
74 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
122 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
156 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
176 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
222 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
258 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
282 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
332 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
368 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
[all …]
H A D_sfdp_pattern_13.py40 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
52 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
74 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
87 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
102 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
126 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_1.py44 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
58 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor, Ignored(), True)
91 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
113 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
130 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor, Ignored(), True)
165 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_5.py45 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
59 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor, Ignored(), True)
93 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
116 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
133 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor, Ignored(), True)
169 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_2.py44 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
58 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_1, Ignored(), True)
91 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
113 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
130 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_1, Ignored(), True)
165 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_6.py47 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
66 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
101 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
126 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
148 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
185 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_3.py46 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
65 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
99 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
123 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
145 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
181 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_4.py46 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
65 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_5, Ignored(), True)
99 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
123 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
145 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_5, Ignored(), True)
181 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_11.py48 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
64 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor, Ignored(), True)
104 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
132 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
151 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor, Ignored(), True)
193 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_10.py48 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
67 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor, Ignored(), True)
106 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
135 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
153 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor, Ignored(), True)
194 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_8.py48 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
67 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor, Ignored(), True)
106 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
135 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
153 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor, Ignored(), True)
194 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_14.py49 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
65 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor, Ignored(), True)
107 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
136 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
155 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor, Ignored(), True)
199 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_7.py50 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
74 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
114 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
145 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
168 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
210 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_19.py50 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
69 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
110 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
137 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
158 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
200 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_9.py50 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
74 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
114 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
145 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
168 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
210 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_12.py50 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
71 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
112 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
142 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
166 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
209 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_15.py52 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
69 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor, Ignored(), True)
116 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
148 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
168 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor, Ignored(), True)
217 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
H A D_sfdp_pattern_17.py54 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
76 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
124 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
158 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
183 sum_dim_IntList_1 = CallFunction(aten.sum.dim_IntList, mul_Tensor_4, Ignored(), True)
233 sum_dim_IntList = CallFunction(aten.sum.dim_IntList, exp_default, Ignored(), True)
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_experimental.py231 sum_1 = torch.ops.aten.sum.dim_IntList(mul, []); mul = None
242 sum_2 = torch.ops.aten.sum.dim_IntList(mul_1, [0], True)
248 sum_3 = torch.ops.aten.sum.dim_IntList(mul_3, [0], True)
255 sum_4 = torch.ops.aten.sum.dim_IntList(view_2, [0], True); view_2 = None
280 sum_1 = torch.ops.aten.sum.dim_IntList(mul, []); mul = None
291 sum_2 = torch.ops.aten.sum.dim_IntList(mul_1, [0], True)
297 sum_3 = torch.ops.aten.sum.dim_IntList(mul_3, [0], True)
304 sum_4 = torch.ops.aten.sum.dim_IntList(view_2, [0], True); view_2 = None
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DSum.cpp20 "Vulkan sum.dim_IntList supports 1d, 2d, 3d, 4d tensors as input!"); in sum_dim()
97 "Vulkan sum.dim_IntList without a dim arg is not implemented"); in sum_dim_IntList()
107 "Vulkan sum.dim_IntList dimension out of range expected to be in range of [", in sum_dim_IntList()
153 TORCH_SELECTIVE_NAME("aten::sum.dim_IntList"), TORCH_FN(sum_dim_IntList)); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/torch/distributed/tensor/examples/
H A Dcomm_mode_features_example.py381 **aten.sum.dim_IntList
385 **aten.sum.dim_IntList
527 **aten.sum.dim_IntList
531 **aten.sum.dim_IntList
663 **aten.sum.dim_IntList
671 **aten.sum.dim_IntList
/aosp_15_r20/external/executorch/backends/arm/_passes/
H A Ddecompose_meandim_pass.py17 exir_ops.edge.aten.sum.dim_IntList,
23 torch.ops.aten.sum.dim_IntList,
37 sum = sum.dim_IntList(x, dim, keepdim)
/aosp_15_r20/external/executorch/extension/training/
H A DREADME.md159 …%sum_1 : [num_users=1] = call_function[target=torch.ops.aten.sum.dim_IntList](args = (%ne_2, []), …
161 …%sum_2 : [num_users=1] = call_function[target=torch.ops.aten.sum.dim_IntList](args = (%where_1, []…
182 …%sum_3 : [num_users=1] = call_function[target=torch.ops.aten.sum.dim_IntList](args = (%mul, [1], T…
190 …%sum_4 : [num_users=1] = call_function[target=torch.ops.aten.sum.dim_IntList](args = (%sub, [0], T…
201 …%sum_5 : [num_users=1] = call_function[target=torch.ops.aten.sum.dim_IntList](args = (%mul_3, [0],…
/aosp_15_r20/external/executorch/backends/arm/test/ops/
H A Dtest_sum.py55 .check_count({"torch.ops.aten.sum.dim_IntList": 1})
75 .check_count({"torch.ops.aten.sum.dim_IntList": 1})
98 .check_count({"torch.ops.aten.sum.dim_IntList": 1})

1234