Fix clean-up for transformations

Signed-off-by: Roman Kazantsev <[email protected]>
mvafin · Jun 8, 2020 · 9b5e8de · 9b5e8de
1 parent 42b6441
commit 9b5e8de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model-optimizer/extensions/back/SpecialNodesFinalization.py b/model-optimizer/extensions/back/SpecialNodesFinalization.py
@@ -160,7 +160,7 @@ def replace_pattern(graph: Graph, match: dict):
         nodes_to_remove = [result_node.id]
 
         # in case only const data consumer that is the result node, remove the whole sub-graph
-        if len(const_data_node.out_nodes()) == 1:
+        if len(const_node.out_port(0).get_destinations()) == 1:
             nodes_to_remove.append(const_node.id)
             nodes_to_remove.append(const_data_node.id)