diff options
Diffstat (limited to 'src/cpu/kernels/CpuScaleKernel.cpp')
| -rw-r--r-- | src/cpu/kernels/CpuScaleKernel.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/cpu/kernels/CpuScaleKernel.cpp b/src/cpu/kernels/CpuScaleKernel.cpp index 1108c7a78e..3063d8f682 100644 --- a/src/cpu/kernels/CpuScaleKernel.cpp +++ b/src/cpu/kernels/CpuScaleKernel.cpp @@ -123,12 +123,12 @@ static const ScaleKernel available_kernels[] = { "neon_u8_scale", [](const ScaleSelectorData & data) { return data.dt == DataType::U8; }, - REGISTER_INTEGER_NEON(arm_compute::cpu::common_neon_scale<uint8_t>) + REGISTER_INTEGER_NEON(arm_compute::cpu::u8_neon_scale) }, { "neon_s16_scale", [](const ScaleSelectorData & data) { return data.dt == DataType::S16; }, - REGISTER_INTEGER_NEON(arm_compute::cpu::common_neon_scale<int16_t>) + REGISTER_INTEGER_NEON(arm_compute::cpu::s16_neon_scale) }, #endif /* defined(ARM_COMPUTE_ENABLE_NEON) */ }; |