Skip to content

Commit a7f9752

Browse files
author
fengshuai03
committed
fix:change api
1 parent c0b5fbd commit a7f9752

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

paddle/fluid/inference/tensorrt/convert/pool3d_op.cc

+3-3
Original file line numberDiff line numberDiff line change
@@ -56,15 +56,15 @@ inline void DealCeilMode(const nvinfer1::Dims &input_shape,
5656
1;
5757

5858
if (floor_d_output_size != ceil_d_output_size) {
59-
post_pad->c() = strides[0] - 1;
59+
post_pad->d[0] = strides[0] - 1;
6060
}
6161

6262
if (floor_h_output_size != ceil_h_output_size) {
63-
post_pad->h() = strides[1] - 1;
63+
post_pad->d[1] = strides[1] - 1;
6464
}
6565

6666
if (floor_w_output_size != ceil_w_output_size) {
67-
post_pad->w() = strides[2] - 1;
67+
post_pad->d[2] = strides[2] - 1;
6868
}
6969
}
7070

0 commit comments

Comments
 (0)