From 1ae556a426a66e554ee080d6ee9f8e04a126b698 Mon Sep 17 00:00:00 2001 From: Maxim Vafin Date: Tue, 14 Dec 2021 00:50:25 +0300 Subject: [PATCH] Fix network -> model renaming merge conflict (#9191) --- tools/mo/openvino/tools/mo/back/offline_transformations.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/mo/openvino/tools/mo/back/offline_transformations.py b/tools/mo/openvino/tools/mo/back/offline_transformations.py index e3796a62d75..ff4f3ac8bb0 100644 --- a/tools/mo/openvino/tools/mo/back/offline_transformations.py +++ b/tools/mo/openvino/tools/mo/back/offline_transformations.py @@ -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()