Searched refs:GpuDeviceInfo (Results 1 – 13 of 13) sorted by relevance
150 struct GpuDeviceInfo { struct159 void set_tensorflow_gpu_device_info(GpuDeviceInfo* g) { in set_tensorflow_gpu_device_info() argument163 virtual const GpuDeviceInfo* tensorflow_gpu_device_info() const { in tensorflow_gpu_device_info()268 GpuDeviceInfo* gpu_device_info_ = nullptr;
65 const DeviceBase::GpuDeviceInfo** dev_info, in PrepareCopy()120 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in SetProtoFromGPU()193 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in DeviceToDeviceCopy()260 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in CopyGPUTensorToCPU()305 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in CopyCPUTensorToGPU()428 const DeviceBase::GpuDeviceInfo* dev_info = nullptr; in CopyGPUTensorToSameGPU()
50 const DeviceBase::GpuDeviceInfo* gpu_info = in ThenExecute()
150 GpuDeviceInfo* gpu_device_info_ = nullptr;
394 gpu_device_info_ = new GpuDeviceInfo; in Init()
104 const DeviceBase::GpuDeviceInfo* dev_info = in MemCpyAsync()110 const DeviceBase::GpuDeviceInfo* dev_info = in MemCpyAsync()
54 const GpuDeviceInfo* tensorflow_gpu_device_info() const override { in tensorflow_gpu_device_info()
135 const DeviceBase::GpuDeviceInfo* gpu_info = in RunAsyncParts()
171 const DeviceBase::GpuDeviceInfo* gpu_info = in RunAsyncParts()
253 const DeviceBase::GpuDeviceInfo* gpu_device_info = in TensorDebugString()
235 std::unique_ptr<GpuDeviceInfo> gpu_device_info_ GUARDED_BY(mu_);
339 auto gpu_device_info = absl::make_unique<GpuDeviceInfo>(); in GetDeviceContextLocked()
409 const DeviceBase::GpuDeviceInfo* gpu_device_info = in StartBody()