auto control_type = tf_executor::ControlType::get(island_op.getContext());
    for (auto& execute : executes) {
      if (execute.GetBody().getOperations().size() > 1) {
        mlir::TF::SplitIsland(execute, control_type);
      }
    }
  }

  island_op.erase();
}

void ParallelExecuteToIslandsPass::runOnOperation() {
  // Find islands with a single `tf_device.parallel_execute` and create

    auto control_type = tf_executor::ControlType::get(island_op.getContext());
    for (auto& replica : replicas) {
      if (replica.GetBody().getOperations().size() > 1) {
        mlir::TF::SplitIsland(replica, control_type);
      }
    }
  }

  island_op.erase();
  return success();
}

void ReplicateToIslandPass::runOnOperation() {