Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h272 virtual bool DeviceMemoryUsage(int64 *free, int64 *total) const { in DeviceMemoryUsage() function
Dstream_executor_pimpl.cc813 bool StreamExecutor::DeviceMemoryUsage(int64 *free, int64 *total) const { in DeviceMemoryUsage() function in stream_executor::StreamExecutor
/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/host/
Dhost_gpu_executor.cc62 bool HostExecutor::DeviceMemoryUsage(int64 *free, int64 *total) const { in DeviceMemoryUsage() function in stream_executor::host::HostExecutor
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor.cc211 bool TpuExecutor::DeviceMemoryUsage(int64* free, int64* total) const { in DeviceMemoryUsage() function in tensorflow::tpu::TpuExecutor
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor_test_util.cc35 TF_Bool DeviceMemoryUsage(const SP_Device* const device, int64_t* const free, in DeviceMemoryUsage() function
Dstream_executor.cc595 bool DeviceMemoryUsage(int64* free, int64* total) const override { in DeviceMemoryUsage() function in stream_executor::__anonc4c542a70111::CStreamExecutor
Dstream_executor_test.cc246 TEST_F(StreamExecutorTest, DeviceMemoryUsage) { in TEST_F() argument
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc812 bool GpuExecutor::DeviceMemoryUsage(int64* free, int64* total) const { in DeviceMemoryUsage() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc719 bool GpuExecutor::DeviceMemoryUsage(int64* free, int64* total) const { in DeviceMemoryUsage() function in stream_executor::gpu::GpuExecutor