namespace mlir::TF {

namespace {

// Exhaust search in the block to get all the ops that have data dependency of
// the cluster.
llvm::SetVector<Operation*> GetAllOpsDependOnCluster(
    const Cluster& c,
    const llvm::DenseMap<Operation*, Cluster*>& op_to_cluster_map) {
  llvm::SetVector<Operation*> ops_depend_on_cluster;
  for (Operation& op : *c.ops.front()->getBlock()) {