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

[MO] Result edges naming issue (#12154)

Browse Source 
* the first approach to fixing the Result edge naming issue

* more robust approach, not depending on the split

* out_nodes_ids type change

* missing if clause
This commit is contained in:
Krzysztof Czugala
2022-08-08 08:55:00 +02:00
committed by GitHub
parent 77b5adcc38
commit ef00498385
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tools/mo/openvino/tools/mo/front/output_cut.py
View File

@@ -27,6 +27,7 @@ class OutputCut(FrontReplacementPattern):
for node in graph.get_op_nodes(needs_removal=True):
fw_info = None
in_node = None
out_nodes_ids = {}
for in_port_idx in node.in_edges():
node_idx = node.in_edge(in_port_idx)['in']
if node_idx in node.in_nodes():
@@ -35,9 +36,15 @@ class OutputCut(FrontReplacementPattern):
if fw_info_value:
fw_info = fw_info_value
break
if fw_info is not None and in_node is not None:
for out_idx in in_node.out_nodes():
out_node = in_node.out_node(out_idx)
out_nodes_ids[out_idx] = out_node.id
graph.erase_node(node)
if fw_info is not None and in_node is not None:
for out_idx in in_node.out_nodes():
set_edge_attribute_between_nodes(in_node, in_node.out_node(out_idx),
if node.id == out_nodes_ids[out_idx]:
set_edge_attribute_between_nodes(in_node, in_node.out_node(out_idx),
'fw_tensor_debug_info', fw_info)