Enable 1x1 convolution optimization
authorDmitry Kurtaev <dmitry.kurtaev+github@gmail.com>
Tue, 2 Apr 2019 11:05:17 +0000 (14:05 +0300)
committerDmitry Kurtaev <dmitry.kurtaev+github@gmail.com>
Tue, 2 Apr 2019 11:05:17 +0000 (14:05 +0300)
modules/dnn/src/layers/convolution_layer.cpp

index b872130..7aaa8bc 100644 (file)
@@ -606,7 +606,7 @@ public:
 
             int inpCnAll = input.size[1], width = input.size[3], height = input.size[2];
             int inpCn = inpCnAll / ngroups;
-            p.is1x1_ = kernel == Size(0,0) && pad == Size(0, 0);
+            p.is1x1_ = kernel == Size(1,1) && pad == Size(0, 0);
             p.useAVX = checkHardwareSupport(CPU_AVX);
             p.useAVX2 = checkHardwareSupport(CPU_AVX2);
             p.useAVX512 = CV_CPU_HAS_SUPPORT_AVX512_SKX;