Skip to content

Commit e4e6681

Browse files
committed
Fixed missing set_attr
test=develop
1 parent 7cd2761 commit e4e6681

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)