Home
last modified time | relevance | path

Searched refs:maybe_gpu_device (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_cluster_util.cc279 absl::optional<absl::string_view> maybe_gpu_device; in PickDeviceForXlaImpl() local
288 if (maybe_gpu_device) { in PickDeviceForXlaImpl()
292 maybe_gpu_device = device_name; in PickDeviceForXlaImpl()
308 if (maybe_unknown_device && maybe_gpu_device) { in PickDeviceForXlaImpl()
311 *maybe_gpu_device)); in PickDeviceForXlaImpl()
323 if (maybe_gpu_device) { in PickDeviceForXlaImpl()
324 *out_device_picked = string(*maybe_gpu_device); in PickDeviceForXlaImpl()