Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dxla_cpu_device.cc73 Status status = device->UseGpuDeviceInfo(); in CreateDevices()
Dxla_gpu_device.cc104 Status status = device->UseGpuDeviceInfo(); in CreateDevices()
Dxla_device.h167 Status UseGpuDeviceInfo() LOCKS_EXCLUDED(mu_);
Dxla_device.cc351 Status XlaDevice::UseGpuDeviceInfo() { in UseGpuDeviceInfo() function in tensorflow::XlaDevice