comments

b3602sss · May 24, 2021 · 274997e · 274997e
1 parent c73c285
commit 274997e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc
@@ -148,6 +148,7 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,
       if (desc.HasAttr("strides")) {
         const std::vector<int> strides =
             BOOST_GET_CONST(std::vector<int>, desc.GetAttr("strides"));
+        // there is no issue if strides.size() less than 2
         if (strides.size() > 1) {
           for (size_t i = 0; i < strides.size(); i++) {
             if (strides[i] > 1) return false;
@@ -243,6 +244,7 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,
       if (desc.HasAttr("strides")) {
         const std::vector<int> strides =
             BOOST_GET_CONST(std::vector<int>, desc.GetAttr("strides"));
+        // there is no issue if strides.size() less than 2
         if (strides.size() > 1) {
           for (size_t i = 0; i < strides.size(); i++) {
             if (strides[i] > 1) return false;