Searched refs:TPUDeviceAndHost (Results 1 – 3 of 3) sorted by relevance
45 struct TPUDeviceAndHost { struct46 TPUDeviceAndHost() {} in TPUDeviceAndHost() function47 TPUDeviceAndHost(llvm::StringRef device, llvm::StringRef host) in TPUDeviceAndHost() function58 llvm::SmallVector<llvm::SmallVector<TPUDeviceAndHost, 8>, 8>; argument
205 devices_and_hosts.push_back({TPUDeviceAndHost( in GetFullMeshTPUExecutionDeviceAssignment()377 num_replicas, llvm::SmallVector<TPUDeviceAndHost, 8>( in GetGeneralTPUExecutionDeviceAssignment()378 num_cores_per_replica, TPUDeviceAndHost())); in GetGeneralTPUExecutionDeviceAssignment()
403 llvm::ArrayRef<llvm::SmallVector<tensorflow::TPUDeviceAndHost, 8>> in AssignDevicesToReplicate() argument506 llvm::ArrayRef<llvm::SmallVector<tensorflow::TPUDeviceAndHost, 8>> in AddToParallelExecuteOp() argument587 llvm::ArrayRef<llvm::SmallVector<tensorflow::TPUDeviceAndHost, 8>> in AssignDevicesToReplicatedExecute() argument