fix comments test=develop

6 years ago · 019e8bbed2
parent 722b0a805f
commit 019e8bbed2
1 changed files with 4 additions and 0 deletions
--- a/paddle/fluid/inference/tensorrt/convert/pool2d_op.cc
+++ b/paddle/fluid/inference/tensorrt/convert/pool2d_op.cc
@ -109,6 +109,10 @@ class Pool2dOpConverter : public OpConverter {
    }

    if (pool_type == "max") {
+      // Under ceil mode, the pre_pad and post_pad are used to
+      // record the the padding size. In some ceil mode cases,
+      // we do not need padding, so we initialize the two vars to 0.
+
      nvinfer1::DimsHW pre_pad(0, 0);
      nvinfer1::DimsHW post_pad(0, 0);
      if (ceil_mode) {