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

  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_driver.h 416 static bool GetDeviceTotalMemory(CUdevice device, uint64 *result);
cuda_driver.cc 567 if (GetDeviceTotalMemory(device, &total_memory)) {
    [all...]
cuda_gpu_executor.cc     [all...]

Completed in 101 milliseconds