Consistent exclusion of subgraph inputs when traversing nodes. (#3171)

150250b0 · Chris Sullivan · Sang Ik Lee · 2078ceac · 150250b0
Commit 150250b0 authored Jul 10, 2019 by Chris Sullivan Committed by Sang Ik Lee Jul 10, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

graph_util.cpp src/ngraph/graph_util.cpp +4 -1

No files found.
--- a/src/ngraph/graph_util.cpp
+++ b/src/ngraph/graph_util.cpp
@@ -75,7 +75,10 @@ void ngraph::traverse_nodes(const NodeVector& subgraph_results,

    for (auto r : subgraph_results)
    {
-        stack.push_front(r);
+        if (instances_seen.count(r) == 0)
+        {
+            stack.push_front(r);
+        }
    }

    while (stack.size() > 0)