diff --git a/caffe2/quantization/server/conv_dnnlowp_op.cc b/caffe2/quantization/server/conv_dnnlowp_op.cc index f4dae731dd1..15b71ceb1f2 100644 --- a/caffe2/quantization/server/conv_dnnlowp_op.cc +++ b/caffe2/quantization/server/conv_dnnlowp_op.cc @@ -1289,11 +1289,12 @@ void ConvDNNLowPOp::ConvNHWCCore_( #endif { if (quantize_groupwise_) { - depthwise_2d_per_channel_quantization_same_pad( + depthwise_2d_same_pad( N, H, W, C, + C, stride_h(), stride_w(), // Shouldn't pass 0 if column_offsets_ is empty here because we @@ -1313,27 +1314,28 @@ void ConvDNNLowPOp::ConvNHWCCore_( dnnlowp_get_thread_num(), dnnlowp_get_num_threads()); } else { - depthwise_2d_same_pad( + depthwise_2d_same_pad( N, H, W, C, + C, stride_h(), stride_w(), // Shouldn't pass 0 if column_offsets_ is empty here because we // need zero_point for padding in_qparams_[INPUT].zero_point, reinterpret_cast(Xdata), - FilterQuantizationParams(0).zero_point, + &FilterQuantizationParams(0).zero_point, *Wq_depthwise_packed_, - requantization_params_[0].real_multiplier, + &requantization_params_[0].real_multiplier, out_qparams_.zero_point, Y_uint8_data, // column_offsets_ empty means column_offsets_ are folded into bias column_offsets_->empty() ? nullptr : column_offsets_->data(), b_quantized_data_, ReluFused, - 1.0f, /*act_times_w_scale*/ + nullptr, /*act_times_w_scale*/ dnnlowp_get_thread_num(), dnnlowp_get_num_threads()); }