1 /* 2 * Copyright (c) Meta Platforms, Inc. and affiliates. 3 * All rights reserved. 4 * 5 * This source code is licensed under the BSD-style license found in the 6 * LICENSE file in the root directory of this source tree. 7 */ 8 9 #pragma once 10 11 #include <executorch/runtime/kernel/kernel_includes.h> 12 13 namespace torch { 14 namespace executor { 15 16 bool check_gelu_args(const Tensor& in, string_view approximate, Tensor& out); 17 18 bool check_glu_args(const Tensor& in, int64_t dim, Tensor& out); 19 20 bool check_log_softmax_args( 21 const Tensor& in, 22 int64_t dim, 23 bool half_to_float, 24 Tensor& out); 25 26 bool check_softmax_args( 27 const Tensor& in, 28 int64_t dim, 29 bool half_to_float, 30 Tensor& out); 31 32 Error resize_glu_out(const Tensor& in, int64_t dim, Tensor& out); 33 34 } // namespace executor 35 } // namespace torch 36