Home
last modified time | relevance | path

Searched defs:DeviceMemoryUsage (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h232 virtual bool DeviceMemoryUsage(int64 *free, int64 *total) const { in DeviceMemoryUsage() function
Dstream_executor_pimpl.cc706 bool StreamExecutor::DeviceMemoryUsage(int64 *free, int64 *total) const { in DeviceMemoryUsage() function in perftools::gputools::StreamExecutor
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h146 bool DeviceMemoryUsage(int64 *free, int64 *total) const override { in DeviceMemoryUsage() function
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h164 bool DeviceMemoryUsage(int64 *free, int64 *total) const override { in DeviceMemoryUsage() function
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc783 bool CUDAExecutor::DeviceMemoryUsage(int64 *free, int64 *total) const { in DeviceMemoryUsage() function in perftools::gputools::cuda::CUDAExecutor