Method UseGpuDeviceInfo

tensorflow/compiler/jit/xla_device.cc:409–413

}

Status XlaDevice::UseGpuDeviceInfo() {
  mutex_lock lock(mu_);
  use_gpu_device_info_ = true;
  return GetDeviceContextLocked().status();
}

Status XlaDevice::TryGetDeviceContext(DeviceContext** out_context) {
  mutex_lock lock(mu_);