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

  /external/tensorflow/tensorflow/stream_executor/
device_description.cc 84 port::HumanReadableNumBytes::ToString(device_memory_size());
device_description.h 118 int64 device_memory_size() const { return device_memory_size_; } function in class:stream_executor::DeviceDescription
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_gpu_executor.cc 1084 uint64 device_memory_size = -1; local
    [all...]
  /external/tensorflow/tensorflow/stream_executor/rocm/
rocm_gpu_executor.cc 921 uint64 device_memory_size = -1; local
    [all...]

Completed in 50 milliseconds