MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / MaybePickDeviceForXla

Function MaybePickDeviceForXla

tensorflow/compiler/jit/device_util.cc:228–234  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

226}
227
228xla::StatusOr<absl::optional<jit::DeviceId>> MaybePickDeviceForXla(
229 const jit::DeviceInfoCache& device_info_cache,
230 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu) {
231 return PickDeviceForXlaImpl(device_info_cache, devices,
232 allow_mixing_unknown_and_cpu,
233 /*failure_to_pick_is_error=*/false);
234}
235} // namespace tensorflow

Callers 2

AreDevicesCompatibleMethod · 0.85
AreDevicesCompatibleMethod · 0.85

Calls 1

PickDeviceForXlaImplFunction · 0.85

Tested by

no test coverage detected