Enable 1x1 convolution optimization

e3286c90 · Dmitry Kurtaev · df1f62b3 · e3286c90
Commit e3286c90 authored Apr 02, 2019 by Dmitry Kurtaev
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

convolution_layer.cpp modules/dnn/src/layers/convolution_layer.cpp +1 -1

No files found.
--- a/modules/dnn/src/layers/convolution_layer.cpp
+++ b/modules/dnn/src/layers/convolution_layer.cpp
@@ -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;