Skip to content

Commit b002761

Browse files
Fix bug that renamed subgraph i/o twice (#1478)
* Fix bug that renamed subgraph i/o twice Signed-off-by: Tom Wildenhain <[email protected]> * Don't rename tensors in subgraphs at all Signed-off-by: Tom Wildenhain <[email protected]>
1 parent eac6b9f commit b002761

File tree

1 file changed

+7
-8
lines changed

1 file changed

+7
-8
lines changed

tf2onnx/tfonnx.py

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -511,8 +511,6 @@ def rename_tensors_in_nodes(onnx_nodes):
511511
for func in ordered_func:
512512
f_inputs_names = [t.name for t in func.inputs]
513513
f_output_names = [t.name for t in func.outputs]
514-
f_inputs_names = rename_tensors_in_list(f_inputs_names)
515-
f_output_names = rename_tensors_in_list(f_output_names)
516514
fg = process_tf_graph(func, continue_on_error, False, target, opset,
517515
custom_op_handlers, custom_rewriter,
518516
extra_opset, shape_override, inputs_as_nchw,
@@ -524,12 +522,13 @@ def rename_tensors_in_nodes(onnx_nodes):
524522

525523
check_io(input_names, output_names, output_shapes)
526524

527-
rename_tensors_in_nodes(onnx_nodes)
528-
input_names = rename_tensors_in_list(input_names)
529-
output_names = rename_tensors_in_list(output_names)
530-
output_shapes = rename_tensors_in_dict(output_shapes)
531-
dtypes = rename_tensors_in_dict(dtypes)
532-
inputs_as_nchw = rename_tensors_in_list(inputs_as_nchw)
525+
if not is_subgraph:
526+
rename_tensors_in_nodes(onnx_nodes)
527+
input_names = rename_tensors_in_list(input_names)
528+
output_names = rename_tensors_in_list(output_names)
529+
output_shapes = rename_tensors_in_dict(output_shapes)
530+
dtypes = rename_tensors_in_dict(dtypes)
531+
inputs_as_nchw = rename_tensors_in_list(inputs_as_nchw)
533532
g = Graph(onnx_nodes, output_shapes, dtypes, target, opset, extra_opset, input_names, output_names, is_subgraph)
534533
g = process_parsed_graph(g, custom_op_handlers, inputs_as_nchw, continue_on_error, custom_rewriter, target,
535534
output_names, initialized_tables, outputs_to_values, outputs_to_dtypes, op_cnt, attr_cnt)

0 commit comments

Comments
 (0)