From a7f9752134c9302b17067ff0d4324e57c73f5be4 Mon Sep 17 00:00:00 2001 From: fengshuai03 Date: Wed, 27 Oct 2021 09:20:45 +0000 Subject: [PATCH] fix:change api --- paddle/fluid/inference/tensorrt/convert/pool3d_op.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/fluid/inference/tensorrt/convert/pool3d_op.cc b/paddle/fluid/inference/tensorrt/convert/pool3d_op.cc index d5a8a7d82fb0f..b8e87a8d94d1f 100644 --- a/paddle/fluid/inference/tensorrt/convert/pool3d_op.cc +++ b/paddle/fluid/inference/tensorrt/convert/pool3d_op.cc @@ -56,15 +56,15 @@ inline void DealCeilMode(const nvinfer1::Dims &input_shape, 1; if (floor_d_output_size != ceil_d_output_size) { - post_pad->c() = strides[0] - 1; + post_pad->d[0] = strides[0] - 1; } if (floor_h_output_size != ceil_h_output_size) { - post_pad->h() = strides[1] - 1; + post_pad->d[1] = strides[1] - 1; } if (floor_w_output_size != ceil_w_output_size) { - post_pad->w() = strides[2] - 1; + post_pad->d[2] = strides[2] - 1; } }