Skip to content

Commit 4c90c40

Browse files
authored
Merge pull request #13641 from kbinias/fix-for-no-set_attr
Fixed missing set_attr
2 parents 6ef6a91 + e4e6681 commit 4c90c40

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/paddle/fluid/transpiler/inference_transpiler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ def _fuse_conv_relu_mkldnn(self, program):
124124
next_op = self.block.ops[i + 1]
125125
if next_op.type == 'relu':
126126
# modify bnorm OP to include relu
127-
current_op.set_attr("fuse_relu", True)
127+
current_op._set_attr("fuse_relu", True)
128128
# remove relu OP
129129
self.block._remove_op(i + 1)
130130
i = i + 1
@@ -454,7 +454,7 @@ def _fuse_conv_eltwise(self, conv_op, eltwise_op):
454454
:type eltwise_op: Operator
455455
'''
456456

457-
conv_op.set_attr("fuse_eltwise", True)
457+
conv_op._set_attr("fuse_eltwise", True)
458458
self.input_map[conv_op.output("Output")[0]] = eltwise_op.input("Y")[0]
459459
self.input_map[eltwise_op.output("Out")[0]] = eltwise_op.input("Y")[0]
460460

0 commit comments

Comments
 (0)