Searched refs:TPUDevicesAndHosts (Results 1 – 2 of 2) sorted by relevance
55 using TPUDevicesAndHosts = variable63 TPUDevicesAndHosts&& tpu_devices) in TPUDeviceAssignment()68 TPUDevicesAndHosts&& tpu_devices, in TPUDeviceAssignment()75 TPUDevicesAndHosts tpu_devices;
187 StatusOr<TPUDevicesAndHosts> GetFullMeshTPUExecutionDeviceAssignment( in GetFullMeshTPUExecutionDeviceAssignment()203 TPUDevicesAndHosts devices_and_hosts; in GetFullMeshTPUExecutionDeviceAssignment()345 StatusOr<std::pair<TPUDevicesAndHosts, xla::DeviceAssignmentProto>>380 TPUDevicesAndHosts devices_and_hosts( in GetGeneralTPUExecutionDeviceAssignment()422 return std::pair<TPUDevicesAndHosts, xla::DeviceAssignmentProto>( in GetGeneralTPUExecutionDeviceAssignment()