Searched refs:TPUDeviceAssignment (Results 1 – 2 of 2) sorted by relevance
61 struct TPUDeviceAssignment { struct62 TPUDeviceAssignment(llvm::StringRef compilation_device, in TPUDeviceAssignment() argument67 TPUDeviceAssignment(llvm::StringRef compilation_device, in TPUDeviceAssignment() argument234 StatusOr<TPUDeviceAssignment> GetTPUCompilationAndExecutionDevices(
449 StatusOr<TPUDeviceAssignment> GetTPUCompilationAndExecutionDevices( in GetTPUCompilationAndExecutionDevices()472 return TPUDeviceAssignment(compilation_device, in GetTPUCompilationAndExecutionDevices()480 return TPUDeviceAssignment(compilation_device, in GetTPUCompilationAndExecutionDevices()