Searched refs:TPUDeviceAssignment (Results 1 – 2 of 2) sorted by relevance
63 struct TPUDeviceAssignment { struct64 TPUDeviceAssignment(llvm::StringRef compilation_device, in TPUDeviceAssignment() argument69 TPUDeviceAssignment(llvm::StringRef compilation_device, in TPUDeviceAssignment() function236 StatusOr<TPUDeviceAssignment> GetTPUCompilationAndExecutionDevices(
445 StatusOr<TPUDeviceAssignment> GetTPUCompilationAndExecutionDevices( in GetTPUCompilationAndExecutionDevices()468 return TPUDeviceAssignment(compilation_device, in GetTPUCompilationAndExecutionDevices()476 return TPUDeviceAssignment(compilation_device, in GetTPUCompilationAndExecutionDevices()