diff --git a/src/operators/kernel/arm/convolution/conv_common.cpp b/src/operators/kernel/arm/convolution/conv_common.cpp index 8db3b36cf43b08c65091035370728db0975709ac..29a9c471aff3e30d92c6de6605fadfde870cb293 100644 --- a/src/operators/kernel/arm/convolution/conv_common.cpp +++ b/src/operators/kernel/arm/convolution/conv_common.cpp @@ -56,10 +56,9 @@ void InitBaseConvKernel(ConvParam *param) { param->Strides()[0] == param->Strides()[1] && param->Dilations()[0] == param->Dilations()[1] && param->Strides()[0] == 1 && param->Dilations()[0] == 1 -#if 0 - && param->Output()->dims()[1] >= 16 && - param->Input()->dims()[1] >= 16 && - param->Input()->dims()[2] <= 140 */ /* refered from ncnn */ +#if 1 + && param->Input()->dims()[1] >= 4 + && param->Input()->dims()[2] >= 16 #endif ) { param->ExecMode() = ConvParam::EXEC_WINOGRAD3X3_FLOAT;