From 5245344d333ec15742008fe46e3f291983620795 Mon Sep 17 00:00:00 2001
From: hjchen2 <chenhoujiangcug@gmail.com>
Date: Sat, 16 Mar 2019 20:22:39 +0800
Subject: [PATCH] Refine winograd conditions for googlenet

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

diff --git a/src/operators/kernel/arm/convolution/conv_common.cpp b/src/operators/kernel/arm/convolution/conv_common.cpp
index 29a9c471af..2a3a5e17e1 100644
--- a/src/operators/kernel/arm/convolution/conv_common.cpp
+++ b/src/operators/kernel/arm/convolution/conv_common.cpp
@@ -57,8 +57,8 @@ void InitBaseConvKernel(ConvParam<CPU> *param) {
                param->Dilations()[0] == param->Dilations()[1] &&
                param->Strides()[0] == 1 && param->Dilations()[0] == 1
 #if 1
-               && param->Input()->dims()[1] >= 4
-               && param->Input()->dims()[2] >= 16
+               && (param->Input()->dims()[1] >= 4 ||
+                   param->Output()->dims()[1] >= 16)
 #endif
     ) {
       param->ExecMode() = ConvParam<CPU>::EXEC_WINOGRAD3X3_FLOAT;
-- 
GitLab