xref: /aosp_15_r20/external/executorch/kernels/optimized/optimized-oss.yaml (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)
1# Copyright (c) Meta Platforms, Inc. and affiliates.
2#
3# This yaml file contains operators that have optimized kernels available.
4# Note that this is a copy of optimized.yaml that does not include gelu and
5# log_softmax, due to the OSS build not currently including sleef.
6# TODO (T183193812)
7
8- op: add.out
9  kernels:
10    - arg_meta: null
11      kernel_name: torch::executor::opt_add_out
12
13- op: add.Scalar_out
14  kernels:
15    - arg_meta: null
16      kernel_name: torch::executor::opt_add_scalar_out
17
18- op: bmm.out
19  kernels:
20    - arg_meta: null
21      kernel_name: torch::executor::opt_bmm_out
22
23- op: div.out
24  kernels:
25    - arg_meta: null
26      kernel_name: torch::executor::opt_div_out
27
28- op: div.Scalar_out
29  kernels:
30    - arg_meta: null
31      kernel_name: torch::executor::opt_div_scalar_out
32
33- op: exp.out
34  kernels:
35    - arg_meta: null
36      kernel_name: torch::executor::opt_exp_out
37
38- op: sigmoid.out
39  kernels:
40    - arg_meta: null
41      kernel_name: torch::executor::opt_sigmoid_out
42
43- op: le.Scalar_out
44  kernels:
45    - arg_meta: null
46      kernel_name: torch::executor::opt_le_scalar_out
47
48- op: le.Tensor_out
49  kernels:
50    - arg_meta: null
51      kernel_name: torch::executor::opt_le_tensor_out
52
53- op: linear.out
54  kernels:
55    - arg_meta: null
56      kernel_name: torch::executor::opt_linear_out
57
58- op: mul.out
59  kernels:
60    - arg_meta: null
61      kernel_name: torch::executor::opt_mul_out
62
63- op: mul.Scalar_out
64  kernels:
65    - arg_meta: null
66      kernel_name: torch::executor::opt_mul_scalar_out
67
68- op: native_layer_norm.out
69  kernels:
70    - arg_meta: null
71      kernel_name: torch::executor::opt_native_layer_norm_out
72
73- op: neg.out
74  kernels:
75    - arg_meta: null
76      kernel_name: torch::executor::opt_neg_out
77
78- op: sub.out
79  kernels:
80    - arg_meta: null
81      kernel_name: torch::executor::opt_sub_out
82
83- op: sub.Scalar_out
84  kernels:
85    - arg_meta: null
86      kernel_name: torch::executor::opt_sub_scalar_out
87