From 8ed3eed7b5481cd6f8819035fb5788971cf033ac Mon Sep 17 00:00:00 2001 From: jignparm Date: Thu, 25 Apr 2019 12:32:28 -0700 Subject: [PATCH] Fix ceil_mode not defined for mkldnn pool.cc warning (#907) --- onnxruntime/core/providers/mkldnn/nn/pool.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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();