Searched refs:gpu_device_info_ (Results 1 – 6 of 6) sorted by relevance
49 gpu_device_info_(gpu_device_info), in IrEmitterContext()64 GpuDeviceInfo gpu_device_info() const { return gpu_device_info_; } in gpu_device_info()92 GpuDeviceInfo gpu_device_info_; variable
163 gpu_device_info_ = g; in set_tensorflow_gpu_device_info()167 return gpu_device_info_; in tensorflow_gpu_device_info()279 GpuDeviceInfo* gpu_device_info_ = nullptr; variable
397 delete gpu_device_info_; in ~BaseGPUDevice()472 gpu_device_info_ = new GpuDeviceInfo; in Init()473 gpu_device_info_->stream = stream_->compute; in Init()474 gpu_device_info_->default_context = device_context_; in Init()475 gpu_device_info_->event_mgr = em_; in Init()479 gpu_device_info_->gpu_id = platform_gpu_id.value(); in Init()480 set_tensorflow_gpu_device_info(gpu_device_info_); in Init()
147 GpuDeviceInfo* gpu_device_info_ = nullptr; variable
258 std::unique_ptr<GpuDeviceInfo> gpu_device_info_ TF_GUARDED_BY(mu_);
369 gpu_device_info_ = std::move(gpu_device_info); in GetDeviceContextLocked()371 << gpu_device_info_.get(); in GetDeviceContextLocked()