Searched refs:DeviceMemoryUsage (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | devices.cc | 67 CHECK(se->DeviceMemoryUsage(&available_memory, &total_memory)); in AvailableGPUMemory()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.h | 164 bool DeviceMemoryUsage(int64 *free, int64 *total) const override { in DeviceMemoryUsage() function
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 146 bool DeviceMemoryUsage(int64 *free, int64 *total) const override { in DeviceMemoryUsage() function
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_internal.h | 271 virtual bool DeviceMemoryUsage(int64 *free, int64 *total) const { in DeviceMemoryUsage() function
|
| D | stream_executor_pimpl.cc | 816 bool StreamExecutor::DeviceMemoryUsage(int64 *free, int64 *total) const { in DeviceMemoryUsage() function in stream_executor::StreamExecutor
817 return implementation_->DeviceMemoryUsage(free, total); in DeviceMemoryUsage()
|
| D | stream_executor_pimpl.h | 361 bool DeviceMemoryUsage(int64 *free, int64 *total) const;
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 194 bool DeviceMemoryUsage(int64* free, int64* total) const override;
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_test.cc | 40 CHECK(se->DeviceMemoryUsage(&available_memory, &total_memory)); in GetTotalGPUMemory()
|
| D | gpu_device.cc | 894 if (!se->DeviceMemoryUsage(&available_memory, &total_memory)) { in SingleVirtualDeviceMemoryLimit()
1537 if (!stream_exec->DeviceMemoryUsage(&free_bytes, &total_bytes)) { in GetValidDeviceIds()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 765 bool GpuExecutor::DeviceMemoryUsage(int64* free, int64* total) const { in DeviceMemoryUsage() function in stream_executor::gpu::GpuExecutor
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 884 bool GpuExecutor::DeviceMemoryUsage(int64* free, int64* total) const { in DeviceMemoryUsage() function in stream_executor::gpu::GpuExecutor
|