diff options
Diffstat (limited to 'src/core/NEON/kernels/NENormalizationLayerKernel.cpp')
-rw-r--r-- | src/core/NEON/kernels/NENormalizationLayerKernel.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/NEON/kernels/NENormalizationLayerKernel.cpp b/src/core/NEON/kernels/NENormalizationLayerKernel.cpp index f6f3d5f238..a409519114 100644 --- a/src/core/NEON/kernels/NENormalizationLayerKernel.cpp +++ b/src/core/NEON/kernels/NENormalizationLayerKernel.cpp @@ -232,7 +232,7 @@ void NENormalizationLayerKernel::normalize_float(const Window &window) }, input, input_squared, output); } -#ifdef ARM_COMPUTE_AARCH64_V8_2 +#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC else if(dt == DataType::F16) { const float16x8_t coeff_vec = vdupq_n_f16(_norm_info.scale_coeff()); @@ -268,7 +268,7 @@ void NENormalizationLayerKernel::normalize_float(const Window &window) }, input, input_squared, output); } -#endif /* ARM_COMPUTE_AARCH64_V8_2 */ +#endif /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */ else { ARM_COMPUTE_ERROR("Not supported"); |