From 05684b7cd0cef308de3a81fc3bc1ab4273998f58 Mon Sep 17 00:00:00 2001 From: Anton Chetverikov Date: Thu, 27 Jan 2022 17:41:58 +0300 Subject: [PATCH] Move tensor name restoring transformation after output normalization (#9904) --- tools/mo/openvino/tools/mo/utils/ir_reader/layer_to_class.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/mo/openvino/tools/mo/utils/ir_reader/layer_to_class.py b/tools/mo/openvino/tools/mo/utils/ir_reader/layer_to_class.py index feb9de4b3bc..30203f5a2ba 100644 --- a/tools/mo/openvino/tools/mo/utils/ir_reader/layer_to_class.py +++ b/tools/mo/openvino/tools/mo/utils/ir_reader/layer_to_class.py @@ -410,12 +410,12 @@ def copy_graph_with_ops(graph: Graph) -> Graph: if op.has_and_set('rt_info'): op.out_node(0)['rt_info'] = op.rt_info - restore_tensor_names(op) - # operations postprocessing with some special types if op.soft_get('type') in postprocessing_op_nodes: postprocessing_op_nodes[op.type](op) + restore_tensor_names(op) + # clean up graph to shape inference new_graph.clean_up()