diff --git a/src/operator/nn/convolution.cc b/src/operator/nn/convolution.cc index 9cc0914e1f01..2c010674314d 100644 --- a/src/operator/nn/convolution.cc +++ b/src/operator/nn/convolution.cc @@ -401,7 +401,7 @@ There are other options to tune the performance. .set_attr("FInferShape", ConvolutionShape) .set_attr("FInferType", ConvolutionType) .set_attr("FInferStorageType", [](const nnvm::NodeAttrs& attrs, - const int dev_mask, DispatchMode* dispatch_mode, + const int dev_mask, DispatchMode* dispatch_mode, std::vector *in_attrs, std::vector *out_attrs) { const ConvolutionParam& params = nnvm::get(attrs.parsed); if (params.no_bias)