Fix a small bug.

revert-3824-remove_grad_op_type
hedaoyuan 8 years ago
parent 5df384d67f
commit 168707cadd

@ -66,7 +66,11 @@ bool ExpandConvLayer::init(const LayerMap &layerMap,
// If depth wise convolution and useGpu == false and ARM-NEON // If depth wise convolution and useGpu == false and ARM-NEON
if (!useGpu_ && isDepthwiseConv(channels_[i], groups_[i]) && !isDeconv_) { if (!useGpu_ && isDepthwiseConv(channels_[i], groups_[i]) && !isDeconv_) {
#if defined(__ARM_NEON__) || defined(__ARM_NEON) #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 #endif
} }

Loading…
Cancel
Save