未验证 提交 07d8f97b 编写于 作者: C cc 提交者: GitHub

Not fuse two scale in the same time, test=develop (#3528)

上级 3a959784
...@@ -26,7 +26,9 @@ class Eliminator : public FuseBase { ...@@ -26,7 +26,9 @@ class Eliminator : public FuseBase {
public: public:
void BuildPattern() override { void BuildPattern() override {
// the previous op's output need updat // the previous op's output need updat
auto* pre_op = OpNode("preop")->assert_is_not_op_type("conditional_block"); auto* pre_op = OpNode("preop")
->assert_is_not_op_type("conditional_block")
->assert_is_not_op_type("scale");
// TODO(Superjomn) check has only one output // TODO(Superjomn) check has only one output
auto* x = VarNode("x")->assert_is_op_input("scale", "X"); auto* x = VarNode("x")->assert_is_op_input("scale", "X");
auto* scale_op = OpNode("scale", "scale") auto* scale_op = OpNode("scale", "scale")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册