Skip to content

Commit 355d848

Browse files
authored
Merge pull request scailable#31 from Andrechang/master
add copy initializers
2 parents fc68528 + f345b34 commit 355d848

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

sclblonnx/merge.py

+7-1
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,8 @@ def concat(
219219
rename_io: bool = False,
220220
edge_match: [] = None,
221221
rename_edges: bool = False,
222-
_verbose: bool = False,
222+
rename_init: bool = False,
223+
_verbose: bool = True,
223224
**kwargs):
224225
"""
225226
concat concatenates two graphs.
@@ -301,6 +302,11 @@ def concat(
301302
sg1 = postfix_names(sg1, "_sg1", "edge")
302303
sg2 = postfix_names(sg2, "_sg2", "edge")
303304

305+
if rename_init:
306+
_print("Renaming init.", "MSG", (not _verbose))
307+
sg1 = postfix_names(sg1, "_sg1", "init")
308+
sg2 = postfix_names(sg2, "_sg2", "init")
309+
304310
# Paste graphs together:
305311
_print("Pasting graphs.", "MSG", (not _verbose))
306312
g = _paste_graphs(sg1, sg2)

0 commit comments

Comments
 (0)