提交 30e2faee 编写于 作者: W wjj19950828

fixed rename bug

上级 2721567b
......@@ -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.
先完成此消息的编辑!
想要评论请 注册