From de687650d28cedf73742dcb635c14a5aae76a16d Mon Sep 17 00:00:00 2001 From: Roman Kazantsev Date: Fri, 4 Nov 2022 10:40:59 +0300 Subject: [PATCH] [TF FE] Fix output names list for meta format (#13828) Signed-off-by: Kazantsev, Roman Signed-off-by: Kazantsev, Roman --- tools/mo/openvino/tools/mo/front/tf/loader.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tools/mo/openvino/tools/mo/front/tf/loader.py b/tools/mo/openvino/tools/mo/front/tf/loader.py index a23f8dc3bfc..b3d6e66180e 100644 --- a/tools/mo/openvino/tools/mo/front/tf/loader.py +++ b/tools/mo/openvino/tools/mo/front/tf/loader.py @@ -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)