xref: /aosp_15_r20/external/pytorch/test/edge/selected_operators.yaml (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1build_features: []
2custom_classes: []
3include_all_non_op_selectives: false
4include_all_operators: false
5kernel_metadata: {}
6et_kernel_metadata:
7  custom::add_3.out:
8    - v1/6;0,1,2,3|6;0,1,2,3|6;0,1,2,3
9    - v1/3;0,1,2,3|3;0,1,2,3|3;0,1,2,3
10  aten::add.out:
11    - v1/6;0,1,2,3|6;0,1,2,3|6;0,1,2,3
12    - v1/3;0,1,2,3|3;0,1,2,3|3;0,1,2,3
13operators:
14  aten::_fake_quantize_per_tensor_affine_cachemask_tensor_qparams.out:
15    debug_info:
16    - functions.yaml
17    include_all_overloads: false
18    is_root_operator: true
19    is_used_for_training: true
20  aten::_reshape_alias_copy.out:
21    debug_info:
22    - functions.yaml
23    include_all_overloads: false
24    is_root_operator: true
25    is_used_for_training: true
26  aten::_softmax.out:
27    debug_info:
28    - functions.yaml
29    include_all_overloads: false
30    is_root_operator: true
31    is_used_for_training: true
32  aten::_to_copy.out:
33    debug_info:
34    - functions.yaml
35    include_all_overloads: false
36    is_root_operator: true
37    is_used_for_training: true
38  aten::_unique2.out:
39    debug_info:
40    - functions.yaml
41    include_all_overloads: false
42    is_root_operator: true
43    is_used_for_training: true
44  aten::add.out:
45    debug_info:
46    - functions.yaml
47    include_all_overloads: false
48    is_root_operator: true
49    is_used_for_training: true
50  aten::addmm.out:
51    debug_info:
52    - functions.yaml
53    include_all_overloads: false
54    is_root_operator: true
55    is_used_for_training: true
56  aten::avg_pool2d.out:
57    debug_info:
58    - functions.yaml
59    include_all_overloads: false
60    is_root_operator: true
61    is_used_for_training: true
62  aten::baddbmm.out:
63    debug_info:
64    - functions.yaml
65    include_all_overloads: false
66    is_root_operator: true
67    is_used_for_training: true
68  aten::bitwise_and.Tensor_out:
69    debug_info:
70    - functions.yaml
71    include_all_overloads: false
72    is_root_operator: true
73    is_used_for_training: true
74  aten::bmm.out:
75    debug_info:
76    - functions.yaml
77    include_all_overloads: false
78    is_root_operator: true
79    is_used_for_training: true
80  aten::cat.out:
81    debug_info:
82    - functions.yaml
83    include_all_overloads: false
84    is_root_operator: true
85    is_used_for_training: true
86  aten::clamp.out:
87    debug_info:
88    - functions.yaml
89    include_all_overloads: false
90    is_root_operator: true
91    is_used_for_training: true
92  aten::clone.out:
93    debug_info:
94    - functions.yaml
95    include_all_overloads: false
96    is_root_operator: true
97    is_used_for_training: true
98  aten::constant_pad_nd.out:
99    debug_info:
100    - functions.yaml
101    include_all_overloads: false
102    is_root_operator: true
103    is_used_for_training: true
104  aten::conv1d.out:
105    debug_info:
106    - functions.yaml
107    include_all_overloads: false
108    is_root_operator: true
109    is_used_for_training: true
110  aten::convolution.out:
111    debug_info:
112    - functions.yaml
113    include_all_overloads: false
114    is_root_operator: true
115    is_used_for_training: true
116  aten::cumsum.out:
117    debug_info:
118    - functions.yaml
119    include_all_overloads: false
120    is_root_operator: true
121    is_used_for_training: true
122  aten::detach_copy.out:
123    debug_info:
124    - functions.yaml
125    include_all_overloads: false
126    is_root_operator: true
127    is_used_for_training: true
128  aten::div.out:
129    debug_info:
130    - functions.yaml
131    include_all_overloads: false
132    is_root_operator: true
133    is_used_for_training: true
134  aten::embedding.out:
135    debug_info:
136    - functions.yaml
137    include_all_overloads: false
138    is_root_operator: true
139    is_used_for_training: true
140  aten::eq.Scalar_out:
141    debug_info:
142    - functions.yaml
143    include_all_overloads: false
144    is_root_operator: true
145    is_used_for_training: true
146  aten::eq.Tensor_out:
147    debug_info:
148    - functions.yaml
149    include_all_overloads: false
150    is_root_operator: true
151    is_used_for_training: true
152  aten::exp.out:
153    debug_info:
154    - functions.yaml
155    include_all_overloads: false
156    is_root_operator: true
157    is_used_for_training: true
158  aten::expand_copy.out:
159    debug_info:
160    - functions.yaml
161    include_all_overloads: false
162    is_root_operator: true
163    is_used_for_training: true
164  aten::floor_divide.out:
165    debug_info:
166    - functions.yaml
167    include_all_overloads: false
168    is_root_operator: true
169    is_used_for_training: true
170  aten::gelu.out:
171    debug_info:
172    - functions.yaml
173    include_all_overloads: false
174    is_root_operator: true
175    is_used_for_training: true
176  aten::grid_sampler_2d.out:
177    debug_info:
178    - functions.yaml
179    include_all_overloads: false
180    is_root_operator: true
181    is_used_for_training: true
182  aten::gt.Scalar_out:
183    debug_info:
184    - functions.yaml
185    include_all_overloads: false
186    is_root_operator: true
187    is_used_for_training: true
188  aten::index.Tensor_out:
189    debug_info:
190    - functions.yaml
191    include_all_overloads: false
192    is_root_operator: true
193    is_used_for_training: true
194  aten::index_put.out:
195    debug_info:
196    - functions.yaml
197    include_all_overloads: false
198    is_root_operator: true
199    is_used_for_training: true
200  aten::index_select.out:
201    debug_info:
202    - functions.yaml
203    include_all_overloads: false
204    is_root_operator: true
205    is_used_for_training: true
206  aten::leaky_relu.out:
207    debug_info:
208    - functions.yaml
209    include_all_overloads: false
210    is_root_operator: true
211    is_used_for_training: true
212  aten::linalg_inv_ex.inverse:
213    debug_info:
214    - functions.yaml
215    include_all_overloads: false
216    is_root_operator: true
217    is_used_for_training: true
218  aten::logit.out:
219    debug_info:
220    - functions.yaml
221    include_all_overloads: false
222    is_root_operator: true
223    is_used_for_training: true
224  aten::masked_fill.Scalar_out:
225    debug_info:
226    - functions.yaml
227    include_all_overloads: false
228    is_root_operator: true
229    is_used_for_training: true
230  aten::max.unary_out:
231    debug_info:
232    - functions.yaml
233    include_all_overloads: false
234    is_root_operator: true
235    is_used_for_training: true
236  aten::max_pool2d_with_indices.out:
237    debug_info:
238    - functions.yaml
239    include_all_overloads: false
240    is_root_operator: true
241    is_used_for_training: true
242  aten::mean.out:
243    debug_info:
244    - functions.yaml
245    include_all_overloads: false
246    is_root_operator: true
247    is_used_for_training: true
248  aten::minimum.out:
249    debug_info:
250    - functions.yaml
251    include_all_overloads: false
252    is_root_operator: true
253    is_used_for_training: true
254  aten::mm.out:
255    debug_info:
256    - functions.yaml
257    include_all_overloads: false
258    is_root_operator: true
259    is_used_for_training: true
260  aten::mul.out:
261    debug_info:
262    - functions.yaml
263    include_all_overloads: false
264    is_root_operator: true
265    is_used_for_training: true
266  aten::native_batch_norm.out:
267    debug_info:
268    - functions.yaml
269    include_all_overloads: false
270    is_root_operator: true
271    is_used_for_training: true
272  aten::native_layer_norm.out:
273    debug_info:
274    - functions.yaml
275    include_all_overloads: false
276    is_root_operator: true
277    is_used_for_training: true
278  aten::ne.Scalar_out:
279    debug_info:
280    - functions.yaml
281    include_all_overloads: false
282    is_root_operator: true
283    is_used_for_training: true
284  aten::nonzero.out:
285    debug_info:
286    - functions.yaml
287    include_all_overloads: false
288    is_root_operator: true
289    is_used_for_training: true
290  aten::permute_copy.out:
291    debug_info:
292    - functions.yaml
293    include_all_overloads: false
294    is_root_operator: true
295    is_used_for_training: true
296  aten::pixel_shuffle.out:
297    debug_info:
298    - functions.yaml
299    include_all_overloads: false
300    is_root_operator: true
301    is_used_for_training: true
302  aten::relu.out:
303    debug_info:
304    - functions.yaml
305    include_all_overloads: false
306    is_root_operator: true
307    is_used_for_training: true
308  aten::remainder.Scalar_out:
309    debug_info:
310    - functions.yaml
311    include_all_overloads: false
312    is_root_operator: true
313    is_used_for_training: true
314  aten::repeat.out:
315    debug_info:
316    - functions.yaml
317    include_all_overloads: false
318    is_root_operator: true
319    is_used_for_training: true
320  aten::round.out:
321    debug_info:
322    - functions.yaml
323    include_all_overloads: false
324    is_root_operator: true
325    is_used_for_training: true
326  aten::rsub.Scalar_out:
327    debug_info:
328    - functions.yaml
329    include_all_overloads: false
330    is_root_operator: true
331    is_used_for_training: true
332  aten::select_copy.int_out:
333    debug_info:
334    - functions.yaml
335    include_all_overloads: false
336    is_root_operator: true
337    is_used_for_training: true
338  aten::sigmoid.out:
339    debug_info:
340    - functions.yaml
341    include_all_overloads: false
342    is_root_operator: true
343    is_used_for_training: true
344  aten::slice_copy.Tensor_out:
345    debug_info:
346    - functions.yaml
347    include_all_overloads: false
348    is_root_operator: true
349    is_used_for_training: true
350  aten::softplus.out:
351    debug_info:
352    - functions.yaml
353    include_all_overloads: false
354    is_root_operator: true
355    is_used_for_training: true
356  aten::sort.values:
357    debug_info:
358    - functions.yaml
359    include_all_overloads: false
360    is_root_operator: true
361    is_used_for_training: true
362  aten::split_copy.Tensor_out:
363    debug_info:
364    - functions.yaml
365    include_all_overloads: false
366    is_root_operator: true
367    is_used_for_training: true
368  aten::split_with_sizes_copy.out:
369    debug_info:
370    - functions.yaml
371    include_all_overloads: false
372    is_root_operator: true
373    is_used_for_training: true
374  aten::stack.out:
375    debug_info:
376    - functions.yaml
377    include_all_overloads: false
378    is_root_operator: true
379    is_used_for_training: true
380  aten::sub.out:
381    debug_info:
382    - functions.yaml
383    include_all_overloads: false
384    is_root_operator: true
385    is_used_for_training: true
386  aten::sum.IntList_out:
387    debug_info:
388    - functions.yaml
389    include_all_overloads: false
390    is_root_operator: true
391    is_used_for_training: true
392  aten::tanh.out:
393    debug_info:
394    - functions.yaml
395    include_all_overloads: false
396    is_root_operator: true
397    is_used_for_training: true
398  aten::topk.values:
399    debug_info:
400    - functions.yaml
401    include_all_overloads: false
402    is_root_operator: true
403    is_used_for_training: true
404  aten::transpose_copy.int_out:
405    debug_info:
406    - functions.yaml
407    include_all_overloads: false
408    is_root_operator: true
409    is_used_for_training: true
410  aten::unbind_copy.int_out:
411    debug_info:
412    - functions.yaml
413    include_all_overloads: false
414    is_root_operator: true
415    is_used_for_training: true
416  aten::unsafe_split.Tensor_out:
417    debug_info:
418    - functions.yaml
419    include_all_overloads: false
420    is_root_operator: true
421    is_used_for_training: true
422  aten::unsqueeze_copy.out:
423    debug_info:
424    - functions.yaml
425    include_all_overloads: false
426    is_root_operator: true
427    is_used_for_training: true
428  aten::upsample_bilinear2d.vec_out:
429    debug_info:
430    - functions.yaml
431    include_all_overloads: false
432    is_root_operator: true
433    is_used_for_training: true
434  aten::upsample_nearest2d.out:
435    debug_info:
436    - functions.yaml
437    include_all_overloads: false
438    is_root_operator: true
439    is_used_for_training: true
440  aten::upsample_nearest2d.vec_out:
441    debug_info:
442    - functions.yaml
443    include_all_overloads: false
444    is_root_operator: true
445    is_used_for_training: true
446  aten::view_copy.out:
447    debug_info:
448    - functions.yaml
449    include_all_overloads: false
450    is_root_operator: true
451    is_used_for_training: true
452  aten::zeros_like.out:
453    debug_info:
454    - functions.yaml
455    include_all_overloads: false
456    is_root_operator: true
457    is_used_for_training: true
458  custom::add_3.out:
459    debug_info:
460    - functions.yaml
461    include_all_overloads: false
462    is_root_operator: true
463    is_used_for_training: true
464