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

test=develop
7 years ago · 7c64aa0fdc
parent 1676094697
commit 7c64aa0fdc
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/transpiler/inference_transpiler.py
+++ b/python/paddle/fluid/transpiler/inference_transpiler.py
@ -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(