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