未验证 提交 77af5053 编写于 作者: M mamingjie-China 提交者: GitHub

Update tf_op_mapper_nhwc.py

上级 4ac94c64
...@@ -74,7 +74,7 @@ class TFOpMapperNHWC(OpMapper): ...@@ -74,7 +74,7 @@ class TFOpMapperNHWC(OpMapper):
unsupported_ops = set() unsupported_ops = set()
sys.stderr.write("Total nodes: {}\n".format(len(self.graph.topo_sort))) sys.stderr.write("Total nodes: {}\n".format(len(self.graph.topo_sort)))
for i, node_name in enumerate(self.graph.topo_sort): for i, node_name in enumerate(self.graph.topo_sort):
sys.stderr.write("\rConverting node {} ... ".format(i)) sys.stderr.write("\rConverting node {} ... ".format(i+1))
node = self.graph.get_node(node_name) node = self.graph.get_node(node_name)
op = node.layer_type op = node.layer_type
if op in self.directly_map_ops: if op in self.directly_map_ops:
...@@ -99,7 +99,7 @@ class TFOpMapperNHWC(OpMapper): ...@@ -99,7 +99,7 @@ class TFOpMapperNHWC(OpMapper):
for op in unsupported_ops: for op in unsupported_ops:
print("========== {} ============".format(op)) print("========== {} ============".format(op))
sys.exit(-1) sys.exit(-1)
sys.stderr.write("\nDone\n") sys.stderr.write("\nDone!\n")
def add_omit_nodes(self, in_node_name, out_node_name): def add_omit_nodes(self, in_node_name, out_node_name):
in_node = self.graph.get_node(in_node_name) in_node = self.graph.get_node(in_node_name)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册