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