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 2021-12-14 00:50:25 +03:00 committed by GitHub
parent 6ad627d60c
commit 1ae556a426
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/mo/openvino/tools/mo/back/offline_transformations.py
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()