diff options
Diffstat (limited to 'src/core/NEON/kernels/NEScaleKernel.cpp')
-rw-r--r-- | src/core/NEON/kernels/NEScaleKernel.cpp | 19 |
1 files changed, 10 insertions, 9 deletions
diff --git a/src/core/NEON/kernels/NEScaleKernel.cpp b/src/core/NEON/kernels/NEScaleKernel.cpp index 1fbd5fb60a..39ed6317a1 100644 --- a/src/core/NEON/kernels/NEScaleKernel.cpp +++ b/src/core/NEON/kernels/NEScaleKernel.cpp @@ -28,7 +28,8 @@ #include "arm_compute/core/utils/misc/Utility.h" #include "src/core/AccessWindowStatic.h" #include "src/core/CPP/Validate.h" -#include "src/core/NEON/kernels/scale/impl/list.h" +#include "src/core/NEON/kernels/scale/impl/NEON/list.h" +#include "src/core/NEON/kernels/scale/impl/SVE/list.h" #include "src/core/NEON/wrapper/wrapper.h" #include "src/core/common/Registrars.h" #include "src/core/helpers/AutoConfiguration.h" @@ -93,15 +94,15 @@ static const ScaleKernel available_kernels[] = #else /* !defined(__ARM_FEATURE_SVE) */ #if defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) { - "fp16_neon_scale", + "common_neon_scale", [](const ScaleSelectorData & data) { return data.dt == DataType::F16; }, - REGISTER_FP16_NEON(arm_compute::cpu::fp16_neon_scale) + REGISTER_FP16_NEON(arm_compute::cpu::common_neon_scale<float16_t>) }, #endif /* !defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) */ { - "f32_neon_scale", + "common_neon_scale", [](const ScaleSelectorData & data) { return data.dt == DataType::F32; }, - REGISTER_FP32_NEON(arm_compute::cpu::fp32_neon_scale) + REGISTER_FP32_NEON(arm_compute::cpu::common_neon_scale<float>) }, { "qasymm8_neon_scale", @@ -114,14 +115,14 @@ static const ScaleKernel available_kernels[] = REGISTER_QASYMM8_SIGNED_NEON(arm_compute::cpu::qasymm8_signed_neon_scale) }, { - "u8_neon_scale", + "common_neon_scale", [](const ScaleSelectorData & data) { return data.dt == DataType::U8; }, - REGISTER_INTEGER_NEON(arm_compute::cpu::u8_neon_scale) + REGISTER_INTEGER_NEON(arm_compute::cpu::common_neon_scale<uint8_t>) }, { - "s16_neon_scale", + "common_neon_scale", [](const ScaleSelectorData & data) { return data.dt == DataType::S16; }, - REGISTER_INTEGER_NEON(arm_compute::cpu::s16_neon_scale) + REGISTER_INTEGER_NEON(arm_compute::cpu::common_neon_scale<int16_t>) }, #endif /* !defined(__ARM_FEATURE_SVE) */ }; |