提交 7c64aa0f 编写于 作者: T Tomasz Patejko

MKLDNN conv + elementwise_add fusion: _set_attr corrected in residual connection fusion

test=develop
上级 16760946
......@@ -465,7 +465,7 @@ class InferenceTranspiler(object):
in_var = self.block.vars[conv_op.input("Input")[0]]
bias_var = self.block.vars[conv_op.input("Bias")[0]]
conv_op.set_attr("fuse_residual_connection", True)
conv_op._set_attr("fuse_residual_connection", True)
attrs = {name: conv_op.attr(name) for name in conv_op.attr_names}
self.block._insert_op(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册