Searched defs:TPUDeviceAndHost (Results 1 – 2 of 2) sorted by relevance
45 struct TPUDeviceAndHost { struct46 TPUDeviceAndHost() {} in TPUDeviceAndHost() function47 TPUDeviceAndHost(llvm::StringRef device, llvm::StringRef host) in TPUDeviceAndHost() argument58 llvm::SmallVector<llvm::SmallVector<TPUDeviceAndHost, 8>, 8>; argument
403 llvm::ArrayRef<llvm::SmallVector<tensorflow::TPUDeviceAndHost, 8>> in AssignDevicesToReplicate()506 llvm::ArrayRef<llvm::SmallVector<tensorflow::TPUDeviceAndHost, 8>> in AddToParallelExecuteOp()587 llvm::ArrayRef<llvm::SmallVector<tensorflow::TPUDeviceAndHost, 8>> in AssignDevicesToReplicatedExecute()