Skip to content

T
tic
Commit 23ba37d4 by Alexander Pivovarov Committed by Wuwei Lin
Options

pooling.cc improvements (#4767)

parent 4f92cfe5
Showing with 4 additions and 13 deletions
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>()) {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment