From 168707caddf9c0ed67a2d87074a5f05b7a63a5c9 Mon Sep 17 00:00:00 2001 From: hedaoyuan Date: Wed, 30 Aug 2017 11:35:19 +0800 Subject: [PATCH] Fix a small bug. --- paddle/gserver/layers/ExpandConvLayer.cpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/paddle/gserver/layers/ExpandConvLayer.cpp b/paddle/gserver/layers/ExpandConvLayer.cpp index 0e84581769..20de475fc3 100644 --- a/paddle/gserver/layers/ExpandConvLayer.cpp +++ b/paddle/gserver/layers/ExpandConvLayer.cpp @@ -66,7 +66,11 @@ bool ExpandConvLayer::init(const LayerMap &layerMap, // If depth wise convolution and useGpu == false and ARM-NEON if (!useGpu_ && isDepthwiseConv(channels_[i], groups_[i]) && !isDeconv_) { #if defined(__ARM_NEON__) || defined(__ARM_NEON) - convType = "NeonDepthwiseConv"; + if ((filterSize_[i] == filterSizeY_[i]) && + (filterSize_[i] == 3 || filterSize_[i] == 4) && + (stride_[i] == strideY_[i]) && (stride_[i] == 1 || stride_[i] == 2)) { + convType = "NeonDepthwiseConv"; + } #endif } -- GitLab