Searched defs:TPUDeviceAndHost (Results 1 – 2 of 2) sorted by relevance
43 struct TPUDeviceAndHost { struct44 TPUDeviceAndHost() {} in TPUDeviceAndHost() function45 TPUDeviceAndHost(llvm::StringRef device, llvm::StringRef host) in TPUDeviceAndHost() function56 llvm::SmallVector<llvm::SmallVector<TPUDeviceAndHost, 8>, 8>; argument
413 llvm::ArrayRef<llvm::SmallVector<tensorflow::TPUDeviceAndHost, 8>> in AssignDevicesToReplicate()469 llvm::ArrayRef<llvm::SmallVector<tensorflow::TPUDeviceAndHost, 8>> in BuildParallelExecuteOp()544 llvm::ArrayRef<llvm::SmallVector<tensorflow::TPUDeviceAndHost, 8>> in AssignDevicesToReplicatedExecute()