Home
last modified time | relevance | path

Searched refs:cuda_memory_usage_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dprofiler_legacy.h67 cuda_memory_usage_(cuda_memory_usage), in cpu_ns_()
143 cuda_memory_usage_ = alloc_size; in updateMemoryStats()
158 return cuda_memory_usage_; in cudaMemoryUsage()
257 int64_t cuda_memory_usage_ = 0; member
H A Dprofiler_legacy.cpp547 eventIValueList.emplace_back(static_cast<int64_t>(cuda_memory_usage_)); in toIValue()