Searched refs:gpu_device_info (Results 1 – 3 of 3) sorted by relevance
339 auto gpu_device_info = absl::make_unique<GpuDeviceInfo>(); in GetDeviceContextLocked() local340 gpu_device_info->stream = stream_.get(); in GetDeviceContextLocked()341 gpu_device_info->default_context = device_context_; in GetDeviceContextLocked()342 set_tensorflow_gpu_device_info(gpu_device_info.get()); in GetDeviceContextLocked()343 gpu_device_info_ = std::move(gpu_device_info); in GetDeviceContextLocked()
253 const DeviceBase::GpuDeviceInfo* gpu_device_info = in TensorDebugString() local255 if (gpu_device_info) { in TensorDebugString()258 gpu_device_info->default_context->CopyDeviceTensorToCPU( in TensorDebugString()
409 const DeviceBase::GpuDeviceInfo* gpu_device_info = in StartBody() local413 if (!is_hostmem_dtype && gpu_device_info && in StartBody()