HomeSort by relevance Sort by last modified time
    Searched refs:DeviceMemoryUsage (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/core/grappler/
devices.cc 64 CHECK(se->DeviceMemoryUsage(&available_memory, &total_memory));
  /external/tensorflow/tensorflow/compiler/xla/service/interpreter/
executor.h 164 bool DeviceMemoryUsage(int64 *free, int64 *total) const override {
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_gpu_executor.h 167 bool DeviceMemoryUsage(int64 *free, int64 *total) const override;
cuda_gpu_executor.cc 783 bool CUDAExecutor::DeviceMemoryUsage(int64 *free, int64 *total) const {
    [all...]
  /external/tensorflow/tensorflow/stream_executor/host/
host_gpu_executor.h 146 bool DeviceMemoryUsage(int64 *free, int64 *total) const override {
  /external/tensorflow/tensorflow/stream_executor/
stream_executor_internal.h 232 virtual bool DeviceMemoryUsage(int64 *free, int64 *total) const {
stream_executor_pimpl.cc 706 bool StreamExecutor::DeviceMemoryUsage(int64 *free, int64 *total) const {
707 return implementation_->DeviceMemoryUsage(free, total);
stream_executor_pimpl.h 330 bool DeviceMemoryUsage(int64 *free, int64 *total) const;
    [all...]
  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device.cc 789 if (!se->DeviceMemoryUsage(&available_memory, &total_memory)) {
    [all...]

Completed in 314 milliseconds