xref: /aosp_15_r20/external/pytorch/test/mobile/model_test/model_ops.yaml (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1root_operators:
2  aten::Bool.Tensor: 32
3  aten::Bool.int: 34
4  aten::Float.Scalar: 30
5  aten::Float.Tensor: 20
6  aten::Float.str: 32
7  aten::FloatImplicit: 5
8  aten::Int.Scalar: 60
9  aten::Int.Tensor: 66
10  aten::Int.float: 11
11  aten::Int.str: 20
12  aten::IntImplicit: 37
13  aten::ScalarImplicit: 9
14  aten::__and__.Tensor: 21
15  aten::__and__.bool: 43
16  aten::__and__.int: 2
17  aten::__contains__.int: 6
18  aten::__contains__.int_list: 37
19  aten::__contains__.str: 60
20  aten::__contains__.str_list: 11
21  aten::__derive_index: 106
22  aten::__getitem__.Dict_int: 2
23  aten::__getitem__.Dict_str: 30
24  aten::__getitem__.str: 39
25  aten::__getitem__.t: 328
26  aten::__is__: 80
27  aten::__isnot__: 79
28  aten::__lshift__.int: 2
29  aten::__not__: 32
30  aten::__range_length: 106
31  aten::__rshift__.int: 2
32  aten::__xor__.bool: 16
33  aten::_aminmax: 18
34  aten::_convolution: 27
35  aten::_convolution.deprecated: 3
36  aten::_infer_size: 9
37  aten::_make_per_tensor_quantized_tensor: 1
38  aten::_pack_padded_sequence: 23
39  aten::_pad_packed_sequence: 23
40  aten::_reshape_from_tensor: 16
41  aten::_set_item.int: 7
42  aten::_set_item.str: 315
43  aten::_set_item.t: 34
44  aten::_shape_as_tensor: 16
45  aten::abs: 1
46  aten::acos: 1
47  aten::adaptive_avg_pool1d: 1
48  aten::adaptive_avg_pool2d: 52
49  aten::adaptive_avg_pool3d: 1
50  aten::add: 15
51  aten::add.Scalar: 81
52  aten::add.Tensor: 138
53  aten::add.float: 9
54  aten::add.int: 128
55  aten::add.int_float: 1
56  aten::add.out: 2
57  aten::add.str: 93
58  aten::add.t: 37
59  aten::add_.Scalar: 24
60  aten::add_.Tensor: 57
61  aten::add_.t: 4
62  aten::addcmul: 2
63  aten::addmm: 7
64  aten::all: 14
65  aten::all.dim: 1
66  aten::allclose: 2
67  aten::any: 22
68  aten::append.t: 155
69  aten::arange: 51
70  aten::arange.start: 32
71  aten::arange.start_step: 56
72  aten::argmax: 5
73  aten::as_strided: 16
74  aten::as_tensor.list: 6
75  aten::atan: 8
76  aten::atan2: 1
77  aten::avg_pool1d: 32
78  aten::avg_pool2d: 4
79  aten::backward: 30
80  aten::batch_norm: 24
81  aten::binary_cross_entropy: 16
82  aten::binary_cross_entropy_with_logits: 6
83  aten::bitwise_not: 20
84  aten::bmm: 48
85  aten::broadcast_tensors: 4
86  aten::cat: 180
87  aten::ceil: 7
88  aten::ceil.float: 35
89  aten::chunk: 72
90  aten::clamp: 65
91  aten::clamp_: 19
92  aten::clamp_min: 7
93  aten::clear.str: 2
94  aten::clone: 39
95  aten::coalesce: 2
96  aten::complex: 1
97  aten::conj: 2
98  aten::constant_pad_nd: 16
99  aten::contiguous: 237
100  aten::conv1d: 28
101  aten::conv2d: 15
102  aten::conv_transpose2d.input: 8
103  aten::copy_: 25
104  aten::copy_.Tensor: 69
105  aten::copy_.int: 1
106  aten::cos: 9
107  aten::count_nonzero: 3
108  aten::cross: 1
109  aten::ctc_loss.Tensor: 1
110  aten::cumsum: 19
111  aten::dequantize.list: 1
112  aten::dequantize.self: 51
113  aten::dequantize.tensor: 61
114  aten::detach: 73
115  aten::diagonal: 1
116  aten::dim: 30
117  aten::div: 39
118  aten::div.Scalar: 33
119  aten::div.Tensor: 88
120  aten::div.Tensor_mode: 12
121  aten::div.float: 8
122  aten::div.int: 35
123  aten::div_.Tensor: 13
124  aten::dot: 1
125  aten::dropout: 78
126  aten::embedding: 40
127  aten::embedding_bag.padding_idx: 2
128  aten::empty.memory_format: 36
129  aten::empty_like: 22
130  aten::empty_strided: 5
131  aten::endswith: 1
132  aten::eq: 3
133  aten::eq.Scalar: 34
134  aten::eq.Tensor: 5
135  aten::eq.int: 133
136  aten::eq.int_list: 56
137  aten::eq.str: 80
138  aten::exp: 30
139  aten::exp.float: 5
140  aten::expand: 73
141  aten::expand_as: 7
142  aten::extend.t: 125
143  aten::feature_dropout: 1
144  aten::fft_irfftn: 1
145  aten::fft_rfftn: 1
146  aten::fill_.Scalar: 25
147  aten::fill_.Tensor: 2
148  aten::find: 1
149  aten::flatten.using_ints: 74
150  aten::flip: 2
151  aten::floor: 9
152  aten::floor.float: 2
153  aten::floor_divide: 8
154  aten::floor_divide.Scalar: 37
155  aten::floordiv.int: 50
156  aten::format: 49
157  aten::frobenius_norm.dim: 1
158  aten::full: 18
159  aten::full_like: 18
160  aten::gather: 25
161  aten::ge.Scalar: 6
162  aten::ge.Tensor: 33
163  aten::ge.int: 107
164  aten::gelu: 53
165  aten::get.default_str: 21
166  aten::glu: 31
167  aten::grid_sampler: 7
168  aten::gt.Scalar: 21
169  aten::gt.float: 48
170  aten::gt.float_int: 4
171  aten::gt.int: 152
172  aten::hardsigmoid: 5
173  aten::hardsigmoid_: 2
174  aten::hardswish_: 6
175  aten::hardtanh: 8
176  aten::hardtanh_: 5
177  aten::hstack: 2
178  aten::imag: 1
179  aten::index.Tensor: 53
180  aten::index_fill.int_Scalar: 67
181  aten::index_put_: 23
182  aten::index_select: 100
183  aten::insert.t: 1
184  aten::is_coalesced: 2
185  aten::is_floating_point: 10
186  aten::isfinite: 1
187  aten::isnan: 1
188  aten::isnumeric: 2
189  aten::item: 86
190  aten::items.str: 8
191  aten::join: 1
192  aten::keys.str: 25
193  aten::layer_norm: 97
194  aten::le.Scalar: 3
195  aten::le.Tensor: 17
196  aten::le.float: 2
197  aten::le.int: 18
198  aten::leaky_relu: 1
199  aten::leaky_relu_: 4
200  aten::len.Dict_int: 5
201  aten::len.Tensor: 34
202  aten::len.str: 41
203  aten::len.t: 327
204  aten::linear: 84
205  aten::linspace: 7
206  aten::list.t: 57
207  aten::log: 27
208  aten::log.float: 1
209  aten::log10: 3
210  aten::log1p: 6
211  aten::log_softmax.int: 104
212  aten::logical_and: 6
213  aten::logical_not: 16
214  aten::logical_or: 1
215  aten::logit: 9
216  aten::lower: 22
217  aten::lstm.data: 10
218  aten::lstm.input: 3
219  aten::lt: 3
220  aten::lt.Scalar: 19
221  aten::lt.Tensor: 5
222  aten::lt.float: 48
223  aten::lt.int: 140
224  aten::masked_fill.Scalar: 48
225  aten::masked_fill_.Scalar: 1
226  aten::matmul: 32
227  aten::max: 54
228  aten::max.dim: 60
229  aten::max.other: 12
230  aten::max_pool2d: 11
231  aten::maximum: 8
232  aten::mean: 13
233  aten::mean.dim: 49
234  aten::meshgrid: 2
235  aten::meshgrid.indexing: 2
236  aten::min: 5
237  aten::min.dim: 8
238  aten::min.other: 28
239  aten::minimum: 8
240  aten::mm: 1
241  aten::mse_loss: 4
242  aten::mul.Scalar: 72
243  aten::mul.Tensor: 188
244  aten::mul.float: 8
245  aten::mul.float_int: 4
246  aten::mul.int: 62
247  aten::mul.int_float: 10
248  aten::mul.left_t: 31
249  aten::mul.out: 1
250  aten::mul_.Scalar: 18
251  aten::mul_.Tensor: 9
252  aten::nan_to_num: 3
253  aten::nan_to_num_: 13
254  aten::narrow: 14
255  aten::ne.Scalar: 21
256  aten::ne.Tensor: 4
257  aten::ne.float: 3
258  aten::ne.float_int: 4
259  aten::ne.int: 123
260  aten::ne.int_float: 2
261  aten::ne.int_list: 21
262  aten::ne.str: 3
263  aten::neg: 73
264  aten::neg.int: 54
265  aten::new_zeros: 32
266  aten::nll_loss: 1
267  aten::nll_loss2d: 1
268  aten::nll_loss_nd: 4
269  aten::nonzero: 12
270  aten::norm.Scalar: 2
271  aten::norm.ScalarOpt_dim: 8
272  aten::numel: 38
273  aten::numpy_T: 1
274  aten::one_hot: 2
275  aten::ones: 50
276  aten::ones_like: 27
277  aten::ord: 39
278  aten::pad: 10
279  aten::permute: 135
280  aten::pop.t: 8
281  aten::pow.Tensor_Scalar: 8
282  aten::pow.Tensor_Tensor: 1
283  aten::pow.int: 1
284  aten::pow.int_float: 2
285  aten::quantile.scalar: 1
286  aten::quantize_per_tensor: 105
287  aten::quantize_per_tensor.tensor_qparams: 7
288  aten::quantize_per_tensor_dynamic: 2
289  aten::quantized_lstm.data: 13
290  aten::rad2deg: 1
291  aten::rand: 33
292  aten::randint.low: 4
293  aten::randn_like: 100
294  aten::real: 1
295  aten::reciprocal: 1
296  aten::reflection_pad2d: 2
297  aten::relu: 176
298  aten::relu_: 15
299  aten::remainder.Scalar: 3
300  aten::remainder.int: 50
301  aten::repeat: 49
302  aten::repeat_interleave.self_Tensor: 1
303  aten::replace: 1
304  aten::replication_pad1d: 1
305  aten::replication_pad2d: 1
306  aten::replication_pad3d: 1
307  aten::requires_grad_: 4
308  aten::reshape: 88
309  aten::resize_as_: 1
310  aten::resolve_conj: 12
311  aten::resolve_neg: 12
312  aten::reverse.t: 2
313  aten::round: 1
314  aten::round.Scalar: 18
315  aten::round.float: 1
316  aten::rsub.Scalar: 7
317  aten::scatter_.src: 17
318  aten::scatter_add_: 16
319  aten::select.int: 111
320  aten::selu: 2
321  aten::set_.source_Tensor: 1
322  aten::sigmoid: 176
323  aten::sin: 9
324  aten::size: 155
325  aten::size.int: 172
326  aten::slice.Tensor: 177
327  aten::slice.str: 18
328  aten::slice.t: 141
329  aten::softmax.int: 139
330  aten::softplus: 2
331  aten::sort: 36
332  aten::sparse_coo_tensor.indices: 1
333  aten::sparse_resize_and_clear_: 1
334  aten::split.Tensor: 3
335  aten::split.sizes: 1
336  aten::split.str: 24
337  aten::split_with_sizes: 1
338  aten::sqrt: 1
339  aten::squeeze: 2
340  aten::squeeze.dim: 51
341  aten::stack: 90
342  aten::startswith: 16
343  aten::str: 25
344  aten::strip: 1
345  aten::sub: 36
346  aten::sub.Scalar: 70
347  aten::sub.Tensor: 145
348  aten::sub.float: 3
349  aten::sub.int: 146
350  aten::sub.int_float: 3
351  aten::sub_.Scalar: 1
352  aten::sub_.Tensor: 6
353  aten::sum: 21
354  aten::sum.dim_IntList: 25
355  aten::t: 3
356  aten::tanh: 78
357  aten::tensor: 156
358  aten::tensor.float: 69
359  aten::tensor.int: 82
360  aten::tensor_split.indices: 4
361  aten::to.device: 20
362  aten::to.dtype: 51
363  aten::to.dtype_layout: 79
364  aten::to.other: 2
365  aten::to.prim_Device: 67
366  aten::to.prim_dtype: 89
367  aten::topk: 21
368  aten::transpose.int: 82
369  aten::triu: 16
370  aten::true_divide.Tensor: 2
371  aten::trunc_: 7
372  aten::type_as: 33
373  aten::unbind.int: 80
374  aten::unique_consecutive: 2
375  aten::unsqueeze: 82
376  aten::unsqueeze_: 12
377  aten::update.str: 8
378  aten::upsample_bicubic2d.vec: 2
379  aten::upsample_bilinear2d.vec: 10
380  aten::upsample_linear1d.vec: 1
381  aten::upsample_nearest1d.vec: 2
382  aten::upsample_nearest2d: 13
383  aten::upsample_nearest2d.vec: 52
384  aten::upsample_nearest3d.vec: 2
385  aten::upsample_trilinear3d.vec: 1
386  aten::values.int: 3
387  aten::view: 117
388  aten::vstack: 2
389  aten::where.ScalarOther: 3
390  aten::where.self: 16
391  aten::zeros: 159
392  aten::zeros.out: 1
393  aten::zeros_like: 11
394  prepacked::conv2d_clamp_prepack: 2
395  prepacked::conv2d_clamp_run: 41
396  prepacked::conv2d_transpose_clamp_prepack: 1
397  prepacked::conv2d_transpose_clamp_run: 2
398  prepacked::linear_clamp_run: 36
399  prim::ModuleContainerIndex.list: 12
400  prim::NumToTensor.Scalar: 18
401  prim::Print: 1
402  prim::RaiseException: 100
403  prim::TupleIndex: 147
404  prim::TupleUnpack: 235
405  prim::Uninitialized: 76
406  prim::abs: 1
407  prim::device: 43
408  prim::dtype: 43
409  prim::max.float: 25
410  prim::max.int: 25
411  prim::max.self_int: 34
412  prim::min: 20
413  prim::min.float: 3
414  prim::min.int: 81
415  prim::min.self_int: 37
416  prim::unchecked_cast: 204
417  quantized::add: 90
418  quantized::add_relu: 2
419  quantized::batch_norm2d: 1
420  quantized::cat: 4
421  quantized::conv1d: 5
422  quantized::conv2d: 2
423  quantized::conv2d.new: 86
424  quantized::conv2d_prepack: 14
425  quantized::conv2d_relu.new: 83
426  quantized::conv_prepack: 3
427  quantized::conv_transpose2d: 1
428  quantized::embedding_4bit: 18
429  quantized::embedding_byte: 34
430  quantized::hardswish: 1
431  quantized::instance_norm: 2
432  quantized::leaky_relu: 2
433  quantized::linear: 39
434  quantized::linear_dynamic: 91
435  quantized::linear_dynamic_fp16: 78
436  quantized::linear_prepack: 69
437  quantized::linear_prepack_fp16: 67
438  quantized::linear_relu: 7
439  quantized::linear_unpack: 46
440  quantized::linear_unpack_fp16: 46
441  quantized::mul: 5
442  quantized::mul.Scalar: 4
443