未验证 提交 1e6bb8d5 编写于 作者: S sangoly 提交者: GitHub

[Fc Fusion] fix fc fusion duplicative arguments bug test=develop (#2135)

上级 61647c35
......@@ -61,6 +61,8 @@ void FcFuser::InsertNewNode(SSAGraph* graph, const key2nodes_t& matched) {
cpp::OpDesc FcFuser::GenOpDesc(const key2nodes_t& matched) {
cpp::OpDesc op_desc = *matched.at("mul")->stmt()->op_info();
op_desc.mutable_inputs()->clear();
op_desc.mutable_outputs()->clear();
op_desc.SetType("fc");
op_desc.SetInput("Input", {matched.at("x")->arg()->name});
op_desc.SetInput("W", {matched.at("W")->arg()->name});
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册