Searched refs:gpu_device_info_ (Results 1 – 5 of 5) sorted by relevance
160 gpu_device_info_ = g; in set_tensorflow_gpu_device_info()164 return gpu_device_info_; in tensorflow_gpu_device_info()268 GpuDeviceInfo* gpu_device_info_ = nullptr; variable
335 delete gpu_device_info_; in ~BaseGPUDevice()394 gpu_device_info_ = new GpuDeviceInfo; in Init()395 gpu_device_info_->stream = streams_[0]->compute; in Init()396 gpu_device_info_->default_context = device_contexts_[0]; in Init()397 gpu_device_info_->event_mgr = em_.get(); in Init()401 gpu_device_info_->gpu_id = platform_gpu_id.value(); in Init()402 set_tensorflow_gpu_device_info(gpu_device_info_); in Init()
150 GpuDeviceInfo* gpu_device_info_ = nullptr; variable
343 gpu_device_info_ = std::move(gpu_device_info); in GetDeviceContextLocked()345 << gpu_device_info_.get(); in GetDeviceContextLocked()
235 std::unique_ptr<GpuDeviceInfo> gpu_device_info_ GUARDED_BY(mu_);