Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dxla_cpu_device.cc97 Status status = device->UseGpuDeviceInfo(); in CreateDevices()
Dxla_gpu_device.cc159 Status status = device->UseGpuDeviceInfo(); in CreateDevices()
Dxla_device.h178 Status UseGpuDeviceInfo() TF_LOCKS_EXCLUDED(mu_);
Dxla_tpu_device.cc381 Status status = device->UseGpuDeviceInfo(); in CreateDevices()
Dxla_device.cc377 Status XlaDevice::UseGpuDeviceInfo() { in UseGpuDeviceInfo() function in tensorflow::XlaDevice