Home
last modified time | relevance | path

Searched refs:GpuDeviceInfo (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h150 struct GpuDeviceInfo { struct
159 void set_tensorflow_gpu_device_info(GpuDeviceInfo* g) { in set_tensorflow_gpu_device_info() argument
163 virtual const GpuDeviceInfo* tensorflow_gpu_device_info() const { in tensorflow_gpu_device_info()
268 GpuDeviceInfo* gpu_device_info_ = nullptr;
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc65 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()
Dgpu_util_platform_specific.cc50 const DeviceBase::GpuDeviceInfo* gpu_info = in ThenExecute()
Dgpu_device.h150 GpuDeviceInfo* gpu_device_info_ = nullptr;
Dgpu_device.cc394 gpu_device_info_ = new GpuDeviceInfo; in Init()
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local.cc104 const DeviceBase::GpuDeviceInfo* dev_info = in MemCpyAsync()
110 const DeviceBase::GpuDeviceInfo* dev_info = in MemCpyAsync()
Drenamed_device.h54 const GpuDeviceInfo* tensorflow_gpu_device_info() const override { in tensorflow_gpu_device_info()
Dring_gatherer.cc135 const DeviceBase::GpuDeviceInfo* gpu_info = in RunAsyncParts()
Dring_reducer.cc171 const DeviceBase::GpuDeviceInfo* gpu_info = in RunAsyncParts()
Dring_alg.cc253 const DeviceBase::GpuDeviceInfo* gpu_device_info = in TensorDebugString()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h235 std::unique_ptr<GpuDeviceInfo> gpu_device_info_ GUARDED_BY(mu_);
Dxla_device.cc339 auto gpu_device_info = absl::make_unique<GpuDeviceInfo>(); in GetDeviceContextLocked()
/external/tensorflow/tensorflow/core/kernels/
Dfunctional_ops.cc409 const DeviceBase::GpuDeviceInfo* gpu_device_info = in StartBody()