Set additional tensor names only if necessary (#9652)

This commit is contained in:
Tomasz Dołbniak authored and GitHub committed 2022-01-14 12:44:50 +01:00
1 parent eb299f819e
commit 8b1514d49e
1 file changed
+3
@@ -47,6 +47,9 @@ def moc_pipeline(argv: argparse.Namespace, moc_front_end: FrontEnd):
:param places An object containing Places and names that will be used for model modification
"""
for new_input in places:
if not hasattr(new_input, 'input_name'):
continue
try:
model.add_name_for_tensor(new_input['node'], new_input['input_name'])
except NotImplementedFailure as e: