Searched defs:TPUDeviceAssignment (Results 1 – 1 of 1) sorted by relevance
63 struct TPUDeviceAssignment { struct64 TPUDeviceAssignment(llvm::StringRef compilation_device, in TPUDeviceAssignment() function69 TPUDeviceAssignment(llvm::StringRef compilation_device, in TPUDeviceAssignment() argument76 std::string compilation_device;77 TPUDevicesAndHosts tpu_devices;78 llvm::Optional<xla::DeviceAssignmentProto> xla_device_assignment;