From 1a4e0390762bc65538767f816f17ca55fd858e1c Mon Sep 17 00:00:00 2001
From: hjchen2 <chenhoujiangcug@gmail.com>
Date: Sat, 16 Mar 2019 19:21:28 +0800
Subject: [PATCH] Set winograd usage conditions to use im2col+gemm for small
 input

---
 src/operators/kernel/arm/convolution/conv_common.cpp | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/operators/kernel/arm/convolution/conv_common.cpp b/src/operators/kernel/arm/convolution/conv_common.cpp
index 8db3b36cf4..29a9c471af 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<CPU> *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<CPU>::EXEC_WINOGRAD3X3_FLOAT;
-- 
GitLab