correct arm convolution depthwise mistakes (Tencent#246)

XiaoguangHu01 · Jan 22, 2018 · 6612178 · 6612178
1 parent 31985b1
commit 6612178
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/layer/arm/convolutiondepthwise_arm.cpp b/src/layer/arm/convolutiondepthwise_arm.cpp
@@ -38,15 +38,15 @@ int ConvolutionDepthWise_arm::forward(const Mat& bottom_blob, Mat& top_blob) con
 
     if (kernel_w != kernel_h || stride_w != stride_h)
     {
-        return Convolution::forward(bottom_blob, top_blob);
+        return ConvolutionDepthWise::forward(bottom_blob, top_blob);
     }
 
     const int kernel_size = kernel_w;
     const int stride = stride_w;
 
     if (kernel_size > 7 || stride > 4 || dilation_w != 1 || dilation_h != 1)
     {
-        return Convolution::forward(bottom_blob, top_blob);
+        return ConvolutionDepthWise::forward(bottom_blob, top_blob);
     }
 
     typedef void (*conv_func)(const Mat&, Mat&, const Mat&, const Mat&);