未验证 提交 e7f5d325 编写于 作者: W wenbin 提交者: GitHub

disable bad case for shuffle pass (#38072)

* disabled bad case

* int to size_t
上级 03b09bd6
......@@ -101,6 +101,21 @@ void ShuffleChannelDetectPass::ApplyImpl(ir::Graph* graph) const {
BOOST_GET_CONST(std::vector<int>, reshape1_desc->GetAttr("shape"));
auto reshape2_shape =
BOOST_GET_CONST(std::vector<int>, reshape2_desc->GetAttr("shape"));
// shuffle_channel dosen't change shape
auto* block = reshape1_desc->Block();
if (block) {
auto x_var_name = reshape1_desc->Input("X")[0];
auto* x_var_desc = block->FindVar(x_var_name);
const auto x_shape = x_var_desc->GetShape();
if (x_shape.size() != reshape2_shape.size()) {
return;
}
for (size_t i = 0; i < x_shape.size(); i++) {
if (x_shape[i] != reshape2_shape[i]) return;
}
}
int i_c = reshape1_shape[2];
int o_c = reshape2_shape[1];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册