未验证 提交 1054b23e 编写于 作者: C csy0225 提交者: GitHub

[XPU] Fix xpu_fuse_pass error caused by weight sharing by other operators. (#51039)

上级 7468bab4
......@@ -76,6 +76,7 @@ FcXPUPattern::FcXPUPattern(PDPattern* pattern,
auto* mul_w = pattern->NewNode(mul_w_repr())
->assert_is_op_input(mul_type_, "Y")
->assert_is_persistable_var()
->assert_has_n_outputs(1)
->assert_more([](Node* node) {
return node->Var()->GetShape().size() == 2;
});
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册