Home
last modified time | relevance | path

Searched defs:resetPeakStats (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/c10/xpu/
H A DXPUCachingAllocator.cpp499 void resetPeakStats() { in resetPeakStats() function in c10::xpu::XPUCachingAllocator::DeviceCachingAllocator
658 void resetPeakStats(DeviceIndex device) { in resetPeakStats() function in c10::xpu::XPUCachingAllocator::XPUAllocator
687 void resetPeakStats(DeviceIndex device) { in resetPeakStats() function
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DCUDAPluggableAllocator.cpp228 void CUDAPluggableAllocator::resetPeakStats(c10::DeviceIndex device) { in resetPeakStats() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAMallocAsyncAllocator.cpp739 void resetPeakStats(c10::DeviceIndex device) override { in resetPeakStats() function
H A DCUDACachingAllocator.h351 inline void resetPeakStats(c10::DeviceIndex device) { in resetPeakStats() function
H A DCUDACachingAllocator.cpp1630 void resetPeakStats() { in resetPeakStats() function in c10::cuda::CUDACachingAllocator::Native::DeviceCachingAllocator
3512 void resetPeakStats(c10::DeviceIndex device) override { in resetPeakStats() function in c10::cuda::CUDACachingAllocator::Native::NativeCachingAllocator