debugging pending_count_

GeorgeVasilakopoulos · Jun 23, 2024 · 1e4e9d9 · 1e4e9d9
1 parent 148299d
commit 1e4e9d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow/core/common_runtime/graph_constructor.cc b/tensorflow/core/common_runtime/graph_constructor.cc
@@ -839,7 +839,7 @@ Status GraphConstructor::InitFromEdges() {
         }
       }
       if (has_loop_back_edge) {
-        pending_count = num_control_edges + 1;
+        pending_count = std::min(num_control_edges + 1, node_def.input_size());
       }
     } else if (IsReturningNode(node_def)) {
       int num_control_edges = 0;
@@ -849,7 +849,7 @@ Status GraphConstructor::InitFromEdges() {
           num_control_edges++;
         }
       }
-      pending_count = num_control_edges + 1;
+      pending_count = std::min(num_control_edges + 1, node_def.input_size());
     } 
     for (int i = 0; i < node_def.input_size(); ++i) {
       StringPiece input_name = node_def.input(i);