diff options
Diffstat (limited to 'src/core/NEON/kernels/NEActivationLayerKernel.cpp')
-rw-r--r-- | src/core/NEON/kernels/NEActivationLayerKernel.cpp | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/src/core/NEON/kernels/NEActivationLayerKernel.cpp b/src/core/NEON/kernels/NEActivationLayerKernel.cpp index a1652447f4..8e91e6b4d1 100644 --- a/src/core/NEON/kernels/NEActivationLayerKernel.cpp +++ b/src/core/NEON/kernels/NEActivationLayerKernel.cpp @@ -533,17 +533,17 @@ typename std::enable_if<std::is_same<T, qasymm8_t>::value, void>::type NEActivat if(act == ActivationFunction::RELU) { tmp = std::max(const_0, in); - tmp = utility::clamp<qasymm8_t>(tmp * s + o); + tmp = utility::clamp<int32_t, qasymm8_t>(tmp * s + o); } else if(act == ActivationFunction::BOUNDED_RELU) { tmp = std::min(a, std::max(const_0, in)); - tmp = utility::clamp<qasymm8_t>(tmp * s + o); + tmp = utility::clamp<int32_t, qasymm8_t>(tmp * s + o); } else if(act == ActivationFunction::LU_BOUNDED_RELU) { tmp = std::min(a, std::max(b, in)); - tmp = utility::clamp<qasymm8_t>(tmp * s + o); + tmp = utility::clamp<int32_t, qasymm8_t>(tmp * s + o); } else if(act == ActivationFunction::LOGISTIC) { @@ -710,17 +710,17 @@ typename std::enable_if<std::is_same<T, qasymm8_signed_t>::value, void>::type NE if(act == ActivationFunction::RELU) { tmp = std::max(const_0, in); - tmp = utility::clamp<qasymm8_signed_t>(tmp * s + o); + tmp = utility::clamp<int32_t, qasymm8_signed_t>(tmp * s + o); } else if(act == ActivationFunction::BOUNDED_RELU) { tmp = std::min(a, std::max(const_0, in)); - tmp = utility::clamp<qasymm8_signed_t>(tmp * s + o); + tmp = utility::clamp<int32_t, qasymm8_signed_t>(tmp * s + o); } else if(act == ActivationFunction::LU_BOUNDED_RELU) { tmp = std::min(a, std::max(b, in)); - tmp = utility::clamp<qasymm8_signed_t>(tmp * s + o); + tmp = utility::clamp<int32_t, qasymm8_signed_t>(tmp * s + o); } else if(act == ActivationFunction::LOGISTIC) { |