From c0ea146176188fe67fee12f5d7ad4c31c20eed14 Mon Sep 17 00:00:00 2001 From: Georgy Krivoruchko Date: Fri, 5 Jul 2024 16:12:52 +0400 Subject: [PATCH] Added a missing change in ir_engine.py --- tools/mo/openvino/tools/mo/utils/ir_engine/ir_engine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/mo/openvino/tools/mo/utils/ir_engine/ir_engine.py b/tools/mo/openvino/tools/mo/utils/ir_engine/ir_engine.py index 08af7e68f551b6..ab748c25c6022b 100644 --- a/tools/mo/openvino/tools/mo/utils/ir_engine/ir_engine.py +++ b/tools/mo/openvino/tools/mo/utils/ir_engine/ir_engine.py @@ -105,7 +105,7 @@ def __load_xml(self): statistics[layer.find('name').text] = {'min': layer.find('min').text, 'max': layer.find('max').text} elif child.tag == 'rt_info': for elem in child: - node_name = elem.attrib['name'] if elem.attrib['name'] else elem.tag + node_name = child.attrib.get('name', child.tag) self.meta_data[node_name] = read_rt_info_attr(elem) # TODO: Remove this part when POT updates to using of rt_info