IntenseWebs/openvino
3
0 Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix network -> model renaming merge conflict (#9191)

Browse Source
This commit is contained in:
Maxim Vafin authored and GitHub committed 2021-12-14 00:50:25 +03:00
1 parent 6ad627d60c
commit 1ae556a426
1 file changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files
tools/mo/openvino/tools/mo/back/offline_transformations.py
+2 -2
View File
@@ -67,7 +67,7 @@ def add_layouts(ov_function, argv: argparse.Namespace):
node_name = list(found)[0]
found_layout = layout_values[node_name]
if found_layout['source_layout']:
prep.input(node_name).network().set_layout(Layout(found_layout['source_layout']))
prep.input(node_name).model().set_layout(Layout(found_layout['source_layout']))
if found_layout['target_layout']:
prep.input(node_name).tensor().set_layout(Layout(found_layout['target_layout']))
@@ -78,7 +78,7 @@ def add_layouts(ov_function, argv: argparse.Namespace):
node_name = list(found)[0]
found_layout = layout_values[node_name]
if found_layout['source_layout']:
prep.output(node_name).network().set_layout(Layout(found_layout['source_layout']))
prep.output(node_name).model().set_layout(Layout(found_layout['source_layout']))
if found_layout['target_layout']:
prep.output(node_name).tensor().set_layout(Layout(found_layout['target_layout']))
prep.build()