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