diff --git a/onnxruntime/core/providers/mkldnn/nn/pool.cc b/onnxruntime/core/providers/mkldnn/nn/pool.cc index 3b48738630..7a0cdcdb7f 100644 --- a/onnxruntime/core/providers/mkldnn/nn/pool.cc +++ b/onnxruntime/core/providers/mkldnn/nn/pool.cc @@ -254,7 +254,7 @@ Status Pool::Compute(OpKernelContext* context) const { strides.assign(kernel_shape.size(), 1); } - std::vector y_dims = PoolBase::SetOutputSize(x_shape, x_shape[1], &pads, this->dilations_, ceil_mode_); + std::vector y_dims = PoolBase::SetOutputSize(x_shape, x_shape[1], &pads, this->dilations_, this->ceil_mode_); Tensor* Y = context->Output(0, TensorShape(y_dims)); size_t num_outputs = OpKernel::Node().OutputDefs().size();