提交 722b0a80 编写于 作者: N nhzlx

fix bug of trt pool

test=develop
上级 d9bb55a1
...@@ -109,8 +109,8 @@ class Pool2dOpConverter : public OpConverter { ...@@ -109,8 +109,8 @@ class Pool2dOpConverter : public OpConverter {
} }
if (pool_type == "max") { if (pool_type == "max") {
nvinfer1::DimsHW pre_pad(paddings[0], paddings[1]); nvinfer1::DimsHW pre_pad(0, 0);
nvinfer1::DimsHW post_pad(paddings[0], paddings[1]); nvinfer1::DimsHW post_pad(0, 0);
if (ceil_mode) { if (ceil_mode) {
// If ceil mode is true, we will pad the appropriate size to the input. // If ceil mode is true, we will pad the appropriate size to the input.
DealCeilMode(input_shape, ksize, strides, paddings, &pre_pad, &post_pad, DealCeilMode(input_shape, ksize, strides, paddings, &pre_pad, &post_pad,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册