| 216 | } |
| 217 | |
| 218 | xla::StatusOr<jit::DeviceId> PickDeviceForXla( |
| 219 | const jit::DeviceInfoCache& device_info_cache, |
| 220 | const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu) { |
| 221 | TF_ASSIGN_OR_RETURN(absl::optional<jit::DeviceId> device_id, |
| 222 | PickDeviceForXlaImpl(device_info_cache, devices, |
| 223 | allow_mixing_unknown_and_cpu, |
| 224 | /*failure_to_pick_is_error=*/true)); |
| 225 | return *device_id; |
| 226 | } |
| 227 | |
| 228 | xla::StatusOr<absl::optional<jit::DeviceId>> MaybePickDeviceForXla( |
| 229 | const jit::DeviceInfoCache& device_info_cache, |
nothing calls this directly
no test coverage detected