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

Update IRReader with the ResultRenaming flag (#10988)

Browse Source
This commit is contained in:
Nikita Malinin
2022-03-23 13:16:40 +03:00
committed by GitHub
parent eb80b28624
commit b50079143f
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tools/mo/openvino/tools/mo/pipeline/common.py
View File

@@ -195,7 +195,7 @@ def set_default_tensor_names_for_parameters_results(graph: Graph):
def prepare_emit_ir(graph: Graph, data_type: str, output_dir: str, output_model_name: str,
mean_data: [list, None] = None, input_names: list = None, meta_info: dict = None,
use_temporary_path=False, convert_types=False):
use_temporary_path=False, convert_types=False, rename_results=True):
if input_names is None:
input_names = []
if meta_info is None:
@@ -220,7 +220,8 @@ def prepare_emit_ir(graph: Graph, data_type: str, output_dir: str, output_model_
for_graph_and_each_sub_graph_recursively(graph, RemoveUselessConvert().find_and_replace_pattern)
ResultRename().find_and_replace_pattern(graph)
if rename_results:
ResultRename().find_and_replace_pattern(graph)
set_default_tensor_names_for_parameters_results(graph)
for sub_graph in [graph] + collect_sub_graphs(graph):

4
tools/mo/openvino/tools/mo/utils/ir_reader/restore_graph.py
View File

@@ -41,7 +41,7 @@ def restore_graph_from_ir(path_to_xml: str, path_to_bin: str = None) -> (Graph,
return new_graph, copy(ir.meta_data)
def save_restored_graph(graph: Graph, path: str, meta_data, name=None):
def save_restored_graph(graph: Graph, path: str, meta_data, name=None, rename_results=True):
"""
Function to apply all necessary transforms from back stage to prepare and save restored graph and metadata.
:param graph: Graph to save
@@ -86,4 +86,4 @@ def save_restored_graph(graph: Graph, path: str, meta_data, name=None):
for_graph_and_each_sub_graph_recursively(graph, RemoveConstOps().find_and_replace_pattern)
for_graph_and_each_sub_graph_recursively(graph, CreateConstNodesReplacement().find_and_replace_pattern)
prepare_emit_ir(graph, data_type, path, name, meta_info=meta_data)
prepare_emit_ir(graph, data_type, path, name, meta_info=meta_data, rename_results=rename_results)