Searched refs:TPUDevicesAndHosts (Results 1 – 2 of 2) sorted by relevance
57 using TPUDevicesAndHosts = variable65 TPUDevicesAndHosts&& tpu_devices) in TPUDeviceAssignment()70 TPUDevicesAndHosts&& tpu_devices, in TPUDeviceAssignment()77 TPUDevicesAndHosts tpu_devices;
183 StatusOr<TPUDevicesAndHosts> GetFullMeshTPUExecutionDeviceAssignment( in GetFullMeshTPUExecutionDeviceAssignment()199 TPUDevicesAndHosts devices_and_hosts; in GetFullMeshTPUExecutionDeviceAssignment()341 StatusOr<std::pair<TPUDevicesAndHosts, xla::DeviceAssignmentProto>>376 TPUDevicesAndHosts devices_and_hosts( in GetGeneralTPUExecutionDeviceAssignment()418 return std::pair<TPUDevicesAndHosts, xla::DeviceAssignmentProto>( in GetGeneralTPUExecutionDeviceAssignment()