pooling.cc improvements (apache#4767)

neo-ai · Mar 2, 2020 · c37198f · c37198f
1 parent 7dbe80f
commit c37198f
Showing 1 changed file with 4 additions and 13 deletions.
diff --git a/src/relay/op/nn/pooling.cc b/src/relay/op/nn/pooling.cc
@@ -134,10 +134,7 @@ bool Pool2DRel(const Array<Type>& types,
     return false;
   }
 
-  std::vector<IndexExpr> oshape;
-  for (const auto& e : dshape) {
-    oshape.push_back(e);
-  }
+  std::vector<IndexExpr> oshape(dshape.begin(), dshape.end());
 
   if (dshape[hidx].as<tir::AnyNode>()) {
     oshape[hidx] = dshape[hidx];
@@ -775,10 +772,7 @@ bool Pool1DRel(const Array<Type>& types,
     return false;
   }
 
-  std::vector<IndexExpr> oshape;
-  for (const auto& e : dshape) {
-    oshape.push_back(e);
-  }
+  std::vector<IndexExpr> oshape(dshape.begin(), dshape.end());
 
   if (dshape[widx].as<tir::AnyNode>()) {
     oshape[widx] = dshape[widx];
@@ -966,12 +960,9 @@ bool Pool3DRel(const Array<Type>& types,
     return false;
   }
 
-  std::vector<IndexExpr> oshape;
-  for (const auto& e : dshape) {
-    oshape.push_back(e);
-  }
+  std::vector<IndexExpr> oshape(dshape.begin(), dshape.end());
 
-  std::vector<int> idxes = {didx, hidx, widx};
+  int idxes[3] = {didx, hidx, widx};
   for (int i = 0; i < 3; i++) {
     int ii = idxes[i];
     if (dshape[ii].as<tir::AnyNode>()) {