diff options
Diffstat (limited to 'src/core/NEON/kernels/detail/NEActivationFunctionDetail.h')
| -rw-r--r-- | src/core/NEON/kernels/detail/NEActivationFunctionDetail.h | 36 |
1 files changed, 18 insertions, 18 deletions
diff --git a/src/core/NEON/kernels/detail/NEActivationFunctionDetail.h b/src/core/NEON/kernels/detail/NEActivationFunctionDetail.h index eef1be06eb..95cdc8f2f9 100644 --- a/src/core/NEON/kernels/detail/NEActivationFunctionDetail.h +++ b/src/core/NEON/kernels/detail/NEActivationFunctionDetail.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2020 Arm Limited. + * Copyright (c) 2018-2021, 2023 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -24,6 +24,8 @@ #ifndef ARM_COMPUTE_DETAIL_NEACTIVATION_FUNCTION_DETAIL_H #define ARM_COMPUTE_DETAIL_NEACTIVATION_FUNCTION_DETAIL_H +#include "arm_compute/function_info/ActivationLayerInfo.h" + #include "src/core/NEON/wrapper/wrapper.h" namespace arm_compute @@ -34,7 +36,7 @@ namespace detail template <typename T, int S> struct dummy { - /** NEON vector type. */ + /** SIMD vector type. */ using ExactType = typename wrapper::traits::neon_vector<T, S>::type; /** Construct a dummy activation object. @@ -68,9 +70,9 @@ struct dummy template <typename T, int S> struct linear { - /** NEON vector type. */ + /** SIMD vector type. */ using ExactType = typename wrapper::traits::neon_vector<T, S>::type; - /** NEON vector tag type. */ + /** SIMD vector tag type. */ using ExactTagType = typename wrapper::traits::neon_vector<T, S>::tag_type; /** Construct a Linear activation object. @@ -112,9 +114,9 @@ struct linear template <typename T, int S> struct square { - /** NEON vector type. */ + /** SIMD vector type. */ using ExactType = typename wrapper::traits::neon_vector<T, S>::type; - /** NEON vector tag type. */ + /** SIMD vector tag type. */ using ExactTagType = typename wrapper::traits::neon_vector<T, S>::tag_type; /** Construct a Square activation object. @@ -148,17 +150,16 @@ struct square template <typename T, int S> struct logistic { - /** NEON vector type. */ + /** SIMD vector type. */ using ExactType = typename wrapper::traits::neon_vector<T, S>::type; - /** NEON vector tag type. */ + /** SIMD vector tag type. */ using ExactTagType = typename wrapper::traits::neon_vector<T, S>::tag_type; /** Construct a Logistic activation object. * * @param[in] act_info Activation layer information. */ - explicit logistic(ActivationLayerInfo act_info) - : vone(wrapper::vdup_n(static_cast<T>(1), ExactTagType{})) + explicit logistic(ActivationLayerInfo act_info) : vone(wrapper::vdup_n(static_cast<T>(1), ExactTagType{})) { ARM_COMPUTE_UNUSED(act_info); } @@ -188,17 +189,16 @@ struct logistic template <typename T, int S> struct relu { - /** NEON vector type. */ + /** SIMD vector type. */ using ExactType = typename wrapper::traits::neon_vector<T, S>::type; - /** NEON vector tag type. */ + /** SIMD vector tag type. */ using ExactTagType = typename wrapper::traits::neon_vector<T, S>::tag_type; /** Construct a RELU activation object. * * @param[in] act_info Activation layer information. */ - explicit relu(ActivationLayerInfo act_info) - : vzero(wrapper::vdup_n(static_cast<T>(0), ExactTagType{})) + explicit relu(ActivationLayerInfo act_info) : vzero(wrapper::vdup_n(static_cast<T>(0), ExactTagType{})) { ARM_COMPUTE_UNUSED(act_info); } @@ -228,9 +228,9 @@ struct relu template <typename T, int S> struct brelu { - /** NEON vector type. */ + /** SIMD vector type. */ using ExactType = typename wrapper::traits::neon_vector<T, S>::type; - /** NEON vector tag type. */ + /** SIMD vector tag type. */ using ExactTagType = typename wrapper::traits::neon_vector<T, S>::tag_type; /** Construct a bounded RELU activation object. @@ -270,9 +270,9 @@ struct brelu template <typename T, int S> struct lubrelu { - /** NEON vector type. */ + /** SIMD vector type. */ using ExactType = typename wrapper::traits::neon_vector<T, S>::type; - /** NEON vector tag type. */ + /** SIMD vector tag type. */ using ExactTagType = typename wrapper::traits::neon_vector<T, S>::tag_type; /** Construct a lower-upper bounded RELU activation object. |