Browse Source

!455 fix bug caused by com

Merge pull request !455 from 于佼良/master
pull/456/MERGE
i-robot Gitee 3 years ago
parent
commit
c86d83adb4
No known key found for this signature in database GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      parser/func_to_graph/func2graph.py

+ 1
- 1
parser/func_to_graph/func2graph.py View File

@@ -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)


Loading…
Cancel
Save