Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dxla_cluster_util.cc254 string* out_device_picked) { in PickDeviceForXlaImpl() argument
270 DCHECK_NE(out_can_pick_device == nullptr, out_device_picked == nullptr); in PickDeviceForXlaImpl()
322 if (out_device_picked) { in PickDeviceForXlaImpl()
324 *out_device_picked = string(*maybe_gpu_device); in PickDeviceForXlaImpl()
326 *out_device_picked = string(*maybe_unknown_device); in PickDeviceForXlaImpl()
328 *out_device_picked = string(*maybe_cpu_device); in PickDeviceForXlaImpl()
339 string* out_device_picked) { in PickDeviceForXla() argument
342 out_device_picked); in PickDeviceForXla()
Dxla_cluster_util.h120 string* out_device_picked);