未验证 提交 58c47e8d 编写于 作者: A Aurelius84 提交者: GitHub

[Cherry-pick][BugFix]Fix set_attr modify underly type (#47500) (#47566)

* Fix set_attr modify underly type (#47500)

* reformat code

* Revert "reformat code"

This reverts commit f11a5d7658633e53c279f11612254937e2d87feb.
上级 7145db6e
......@@ -6106,7 +6106,7 @@ class Program(object):
for j in six.moves.range(block.op_size()):
op = block.op(j)
if op.has_attr('is_test'):
op._set_attr('is_test', True)
op._set_bool_attr('is_test', True)
if op.type() == "batch_norm":
# Remove the output ReserveSpace of batch_norm if exists.
op.remove_output("ReserveSpace")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册