未验证 提交 f50de679 编写于 作者: Y yeliang2258 提交者: GitHub

fix squueze_transpose (#47911)

上级 5478e1a5
......@@ -961,6 +961,7 @@ PDNode *patterns::OperatorActivation::operator()(
PDNode *patterns::Squeeze2Transpose2::operator()() {
auto *squeeze2_op_in = pattern->NewNode(squeeze2_op_in_repr())
->AsInput()
->assert_has_n_outputs(1)
->assert_is_op_input("squeeze2", "X");
auto *squeeze2_op = pattern->NewNode(squeeze2_op_repr())
->assert_is_op("squeeze2")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册