Searched defs:DeviceMemoryUsage (Results 1 – 10 of 10) sorted by relevance
| /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
|
| /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/host/ |
| 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/stream_executor/tpu/ |
| 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/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
|
| D | stream_executor.cc | 582 bool DeviceMemoryUsage(int64_t* free, int64_t* total) const override { in DeviceMemoryUsage() function in stream_executor::__anoncabb94000111::CStreamExecutor
|
| D | stream_executor_test.cc | 248 TEST_F(StreamExecutorTest, DeviceMemoryUsage) { in TEST_F() argument
|
| /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
|