Skip to content
This repository was archived by the owner on Feb 7, 2023. It is now read-only.

Commit e5a9b7f

Browse files
authored
Fix pixel_shuffle_reshape naming convention to avoid duplicate naming for output node name (#481)
Fixes #446
1 parent b6221b4 commit e5a9b7f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

onnx_coreml/_transformers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -481,8 +481,8 @@ def merge(self, graph, nodes): # type: (Graph, Sequence[Node]) -> Sequence[Node
481481
transpose_1.children = []
482482
transpose_1.attrs['perm'] = [0, 3, 1, 2]
483483

484-
reshape_output_name = 'pixel_shuffle_reshape'
485-
transpose_output_name = 'pixel_shuffle_transpose'
484+
reshape_output_name = final_reshape.name + '_pixel_shuffle_reshape'
485+
transpose_output_name = final_reshape.name + '_pixel_shuffle_transpose'
486486

487487
transpose_1.outputs = [
488488
self.get_unique_edge_name(graph, transpose_output_name)

0 commit comments

Comments
 (0)