[&](mlir::tf_device::ClusterOp descendant) { cluster = descendant; });
  llvm::SmallVector<std::string, 8> core_to_host;
  EXPECT_TRUE(mlir::succeeded(GetDeviceToHostMap(cluster, core_to_host)));
  EXPECT_EQ(core_to_host.size(), 2);
  EXPECT_EQ(core_to_host[0], "/job:localhost/replica:0/task:0/device:CPU:0");
  EXPECT_EQ(core_to_host[1], "/job:localhost/replica:0/task:0/device:CPU:0");
}

  int num_cores_per_replica = GetNumCoresPerReplica(cluster);
  llvm::SmallVector<std::string, 8> core_to_host;
  core_to_host.reserve(num_cores_per_replica);
  for (int core = 0; core < num_cores_per_replica; ++core) {
    core_to_host.push_back(GetDeviceAliasForHostOfLogicalCore(core));
  }
  return core_to_host;
}

// Get the map from `core` to host device for a non-replicated TPU cluster.

      SmallVector<std::string, 8> core_to_host;
      if (failed(tensorflow::GetDeviceToHostMap(device_cluster, core_to_host)))
        return signalPassFailure();
      if (failed(CreateParallelExecuteForOutsideCompilation(
              device_cluster, tmp_parallel_execute_ops, is_map_oc, core_to_host,
              tensorflow::HasTPUDevice(devices))))
        return signalPassFailure();

// is the associated virtual device name (`TPU_REPLICATED_HOST_i`).
mlir::LogicalResult GetDeviceToHostMap(
    mlir::tf_device::ClusterOp cluster,
    llvm::SmallVector<std::string, 8>& core_to_host);

// Returns the first TPU device, for use in the non-replicated case. The list of
// TPU devices is retrived from `op`'s module ancestor.
mlir::LogicalResult GetNonReplicatedTPU0(mlir::Operation* op,