GetXlaScope(node));

    cluster_for_node_[node->id()].Get() = new_cluster;
  }

  return absl::OkStatus();
}

absl::StatusOr<bool> IsIdentityDrivingConstsInLoop(Node* node) {
  if (!node->IsIdentity()) {
    return false;
  }

  // Check if the Identity is driven by a Switch on its true path.
  auto it = absl::c_find_if(node->in_edges(), [](const Edge* e) {