[tf.data] Avoid visiting the same node multiple times in ComputeHash.

PiperOrigin-RevId: 335672916 Change-Id: Ib7da5fc41938b772310b058eaf2fb9474bf466e3
2020-10-06 10:45:11 -07:00 · 2020-10-06 10:45:11 -07:00 · d7bb5785ce
commit d7bb5785ce
parent acfb05435f
1 changed files with 3 additions and 0 deletions
--- a/tensorflow/core/kernels/data/hash_utils.cc
+++ b/tensorflow/core/kernels/data/hash_utils.cc
@ -161,6 +161,9 @@ class GraphHasher {
    while (!bfs_queue.empty()) {
      const NodeDef* node = bfs_queue.front();
      bfs_queue.pop();
+      if (visited.contains(node->name())) {
+        continue;
+      }
      visited.insert(node->name());
      NodeRep node_rep;
      for (int i = 0; i < node->input_size(); ++i) {