aboutsummaryrefslogtreecommitdiff
path: root/src/core/NEON/kernels/batchnormalization/impl/NEON/fp16.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/core/NEON/kernels/batchnormalization/impl/NEON/fp16.cpp')
-rw-r--r--src/core/NEON/kernels/batchnormalization/impl/NEON/fp16.cpp11
1 files changed, 9 insertions, 2 deletions
diff --git a/src/core/NEON/kernels/batchnormalization/impl/NEON/fp16.cpp b/src/core/NEON/kernels/batchnormalization/impl/NEON/fp16.cpp
index dfadef34f7..fd17b98f7b 100644
--- a/src/core/NEON/kernels/batchnormalization/impl/NEON/fp16.cpp
+++ b/src/core/NEON/kernels/batchnormalization/impl/NEON/fp16.cpp
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2020 Arm Limited.
+ * Copyright (c) 2020-2021 Arm Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -138,7 +138,14 @@ namespace cpu
void fp16_neon_batch_normalization(ITensor *src, ITensor *dst, const ITensor *mean, const ITensor *var, const ITensor *beta, const ITensor *gamma,
float epsilon, ActivationLayerInfo &act_info, const Window &window)
{
- fused_map[act_info.activation()](src, dst, mean, var, beta, gamma, epsilon, act_info, window);
+ if(act_info.enabled())
+ {
+ fused_map[act_info.activation()](src, dst, mean, var, beta, gamma, epsilon, act_info, window);
+ }
+ else
+ {
+ batch_normalization<detail::dummy<float16_t, 8>>(src, dst, mean, var, beta, gamma, epsilon, act_info, window);
+ }
}
} // namespace cpu
} // namespace arm_compute