From 559294196381f194a58ccbd90ae417bfaccc05f0 Mon Sep 17 00:00:00 2001 From: xiebaiyuan Date: Thu, 28 Feb 2019 11:18:13 +0800 Subject: [PATCH] use gemm to s1p0 in conv --- .../kernel/central-arm-func/dwconv_bn_relu_arm_func.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/operators/kernel/central-arm-func/dwconv_bn_relu_arm_func.h b/src/operators/kernel/central-arm-func/dwconv_bn_relu_arm_func.h index e0299d00ae..1504850324 100644 --- a/src/operators/kernel/central-arm-func/dwconv_bn_relu_arm_func.h +++ b/src/operators/kernel/central-arm-func/dwconv_bn_relu_arm_func.h @@ -118,7 +118,8 @@ void DWConvBNReluCompute(const FusionDWConvBNReluParam ¶m) { if (param.Groups() == param.Input()->dims()[1] && param.Input()->dims()[1] == param.Output()->dims()[1] && param.Filter()->dims()[2] == param.Filter()->dims()[3] && - param.Filter()->dims()[2] == 3 && param.Strides()[0] == 1) { + param.Filter()->dims()[2] == 3 && param.Strides()[0] == 1 && + param.paddings_[0] == 1) { math::DepthwiseConvAddBNRelu3x3s1p1(param.Input(), param.Filter(), param.Output(), param.NewScale(), param.NewBias(), true); -- GitLab