diff --git a/src/operators/kernel/arm/convolution/conv_kernel.cpp b/src/operators/kernel/arm/convolution/conv_kernel.cpp index 18d1e7e640eb30d027df1e9519bd4f4419142aa1..d73046489d7ee1547a9ad4390f0566bb028fac4c 100644 --- a/src/operators/kernel/arm/convolution/conv_kernel.cpp +++ b/src/operators/kernel/arm/convolution/conv_kernel.cpp @@ -30,10 +30,10 @@ bool ConvKernel::Init(ConvParam *param) { template <> void ConvKernel::Compute(const ConvParam ¶m) { switch (param.ExecMode()) { -#ifndef __aarch64__ case ConvParam::EXEC_GEMM_INT8: GemmConv(param); break; +#ifndef __aarch64__ case ConvParam::EXEC_DEPTHWISE3x3_INT8: DepthwiseConv3x3(param); break; diff --git a/src/operators/kernel/central-arm-func/conv_arm_func.cpp b/src/operators/kernel/central-arm-func/conv_arm_func.cpp index fbdf52df911a57534c5dbf297b351c6b3dc59e87..c033d24b97dddb62ccec7a9ea912d544c6586452 100644 --- a/src/operators/kernel/central-arm-func/conv_arm_func.cpp +++ b/src/operators/kernel/central-arm-func/conv_arm_func.cpp @@ -263,8 +263,8 @@ template void DepthwiseConv3x3(const ConvParam ¶m); template void DepthwiseConv5x5(const ConvParam ¶m); template void SlidingwindowConv3x3(const ConvParam ¶m); -#ifndef __aarch64__ template void GemmConv(const ConvParam ¶m); +#ifndef __aarch64__ template void DepthwiseConv3x3(const ConvParam ¶m); template void DepthwiseConv5x5(const ConvParam ¶m); #endif