From 5c7be85d10e8dea5b9f800e2a4276c6849ad6cf2 Mon Sep 17 00:00:00 2001 From: yu-jiaoliang Date: Wed, 5 Jan 2022 11:33:12 +0800 Subject: [PATCH] fix bug --- parser/func_to_graph/func2graph.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/parser/func_to_graph/func2graph.py b/parser/func_to_graph/func2graph.py index e1a1ea9..c197da0 100644 --- a/parser/func_to_graph/func2graph.py +++ b/parser/func_to_graph/func2graph.py @@ -181,7 +181,7 @@ def convert_function_def_to_graph_def(fdef, input_shapes=None, copy_functions=Tr # Update inputs of all nodes in graph. for node_def in graph_def.node: for i in range(len(node_def.input)): - node_def.input[i] = nested_to_flat_tensor_name[node_def.input.get(i)] + node_def.input[i] = nested_to_flat_tensor_name.get(node_def.input[i]) # Create _Retval for output nodes. create_retval_for_output_nodes(fdef, graph_def, nested_to_flat_tensor_name)