Checked for whether a node exists in the graph before extracting it due to high out-degree.

Change: 135964994

1 files changed, 5 insertions, 2 deletions

diff --git a/tensorflow/tensorboard/components/tf-graph-common/lib/render.ts b/tensorflow/tensorboard/components/tf-graph-common/lib/render.ts
index b4945eed10..3bea0061c6 100644
--- a/tensorflow/tensorboard/components/tf-graph-common/lib/render.ts
+++ b/tensorflow/tensorboard/components/tf-graph-common/lib/render.ts
@@ -1452,8 +1452,11 @@ function extractHighInOrOutDegree(renderNode: RenderGroupNodeInfo) {
   outDegreeUpperBound = Math.max(outDegreeUpperBound, minUpperBound);
   for (let i = validNodeCount - 1;
        nodeToOutDegree[sortedByOutDegree[i]] > outDegreeUpperBound; i--) {
-    if (graph.node(sortedByOutDegree[i]).isInExtract) {
-      // This node has already been extracted due to high in-degree. Ignore it.
+    let node = graph.node(sortedByOutDegree[i]);
+    if (!node || node.isInExtract) {
+      // This node has already been extracted due to high in-degree. It might
+      // have been removed from the graph in general (during in-degree
+      // extraction) due to a lack of neighbors. Do not extract this node twice.
       continue;
     }