Searched defs:TPUDeviceAssignment (Results 1 – 1 of 1) sorted by relevance
61 struct TPUDeviceAssignment { struct62 TPUDeviceAssignment(llvm::StringRef compilation_device, in TPUDeviceAssignment() argument67 TPUDeviceAssignment(llvm::StringRef compilation_device, in TPUDeviceAssignment() argument74 std::string compilation_device;75 TPUDevicesAndHosts tpu_devices;76 llvm::Optional<xla::DeviceAssignmentProto> xla_device_assignment;