未验证 提交 f2e933cf 编写于 作者: J Jason 提交者: GitHub

Merge pull request #820 from wjj19950828/fixed_rename

Fixed rename bug
......@@ -583,6 +583,9 @@ class ONNXDecoder(object):
item.name = self.make_variable_name(item.name)
for node in graph.node:
node.name = node.output[0]
# Avoid topological sort errors caused by :: in the name
if "::" in node.name and len(node.output) > 1:
node.name = node.name.replace('::', '_')
if ":" in node.name and len(
node.output) > 1 and node.op_type != "LSTM":
node.name = node.name.split(':')[0]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册