| /external/tensorflow/tensorflow/core/grappler/ |
| D | devices.cc | 91 CHECK(se->DeviceMemoryUsage(&available_memory, &total_memory)); in AvailableGPUMemory()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test_util.cc | 35 TF_Bool DeviceMemoryUsage(const SP_Device* const device, int64_t* const free, in DeviceMemoryUsage() function
107 se->device_memory_usage = DeviceMemoryUsage; in PopulateDefaultStreamExecutor()
|
| D | stream_executor_test.cc | 248 TEST_F(StreamExecutorTest, DeviceMemoryUsage) { in TEST_F() argument
260 executor->DeviceMemoryUsage(&free, &total); in TEST_F()
|
| D | stream_executor.cc | 582 bool DeviceMemoryUsage(int64_t* free, int64_t* total) const override { in DeviceMemoryUsage() function in stream_executor::__anon7d2d81310111::CStreamExecutor
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | gpu_device.cc | 71 if (!executor->DeviceMemoryUsage(&free_memory, &total_memory)) { in CreateCudaAsyncAllocator()
232 if (!executor->DeviceMemoryUsage(&free_memory, &total_memory)) { in CreateBFCAllocator()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
| D | host_gpu_executor.h | 133 bool DeviceMemoryUsage(int64_t* free, int64_t* total) const override;
|
| D | host_gpu_executor.cc | 62 bool HostExecutor::DeviceMemoryUsage(int64_t* free, int64_t* total) const { in DeviceMemoryUsage() function in stream_executor::host::HostExecutor
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.h | 163 bool DeviceMemoryUsage(int64_t *free, int64_t *total) const override { in DeviceMemoryUsage() function
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | tpu_executor.h | 96 bool DeviceMemoryUsage(int64_t* free, int64_t* total) const override;
|
| D | tpu_executor.cc | 211 bool TpuExecutor::DeviceMemoryUsage(int64_t* free, int64_t* total) const { in DeviceMemoryUsage() function in tensorflow::tpu::TpuExecutor
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_factory.cc | 88 if (!se->DeviceMemoryUsage(&available_memory, &total_memory)) { in SingleVirtualDeviceMemoryLimit()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream_executor_internal.h | 270 virtual bool DeviceMemoryUsage(int64_t* free, int64_t* total) const { in DeviceMemoryUsage() function
|
| D | stream_executor_pimpl.cc | 816 bool StreamExecutor::DeviceMemoryUsage(int64_t* free, int64_t* total) const { in DeviceMemoryUsage() function in stream_executor::StreamExecutor
817 return implementation_->DeviceMemoryUsage(free, total); in DeviceMemoryUsage()
|
| D | stream_executor_pimpl.h | 344 bool DeviceMemoryUsage(int64_t* free, int64_t* total) const;
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/ |
| D | gpu_executor.h | 242 bool DeviceMemoryUsage(int64_t* free, int64_t* total) const override;
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_test.cc | 44 CHECK(se->DeviceMemoryUsage(&available_memory, &total_memory)); in GetTotalGPUMemory()
|
| D | gpu_device.cc | 1093 if (!se->DeviceMemoryUsage(&available_memory, &total_memory)) { in SingleVirtualDeviceMemoryLimit()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 765 if (!stream_executor->DeviceMemoryUsage(&mem_free, &mem_total)) { in Compute()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 731 bool GpuExecutor::DeviceMemoryUsage(int64_t* free, int64_t* total) const { in DeviceMemoryUsage() function in stream_executor::gpu::GpuExecutor
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 863 bool GpuExecutor::DeviceMemoryUsage(int64_t* free, int64_t* total) const { in DeviceMemoryUsage() function in stream_executor::gpu::GpuExecutor
|