diff --git a/mace/kernels/opencl/fused_conv_2d_opencl.cc b/mace/kernels/opencl/fused_conv_2d_opencl.cc index 9e23c57f40aee458d5bc7fd02708456659fc8732..86aa0424f003c4c5815766bbab9dac2e6f5ee191 100644 --- a/mace/kernels/opencl/fused_conv_2d_opencl.cc +++ b/mace/kernels/opencl/fused_conv_2d_opencl.cc @@ -73,9 +73,9 @@ void FusedConv2dFunctor::operator()(const Tensor *input, if (kernel_h == kernel_w && kernel_h <= 5 && selector[kernel_h - 1][strides_[0] - 1] != nullptr) { auto conv2d_func = selector[kernel_h - 1][strides_[0] - 1]; - conv2d_func(input, filter, bias, false, paddings.data(), DataTypeToEnum::value, output); + conv2d_func(input, filter, bias, true, paddings.data(), DataTypeToEnum::value, output); } else { - Conv2dOpencl(input, filter, bias, false, strides_[0], paddings.data(), DataTypeToEnum::value, output); + Conv2dOpencl(input, filter, bias, true, strides_[0], paddings.data(), DataTypeToEnum::value, output); } }