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

[TF FE] Fix output names list for meta format (#13828)

Browse Source 
Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>

Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>
This commit is contained in:
Roman Kazantsev 2022-11-04 10:40:59 +03:00 committed by GitHub
parent 6fc23b4768
commit de687650d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/mo/openvino/tools/mo/front/tf/loader.py
View File

@ -287,11 +287,12 @@ def load_tf_graph_def(graph_file_name: str = "", is_binary: bool = True, checkpo
def convert_to_pb(argv: argparse.Namespace):
from openvino.tools.mo.utils.cli_parser import get_model_name
user_output_node_names_list = argv.output.split(',') if argv.output else None
graph_def, _, _, _ = load_tf_graph_def(
graph_file_name=argv.input_model,
is_binary=not argv.input_model_is_text,
checkpoint=argv.input_checkpoint,
user_output_node_names_list=argv.output,
user_output_node_names_list=user_output_node_names_list,
model_dir=argv.saved_model_dir,
meta_graph_file=argv.input_meta_graph,
saved_model_tags=argv.saved_model_tags)