Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/detail/
H A DMPSHooksInterface.h64 virtual void setMemoryFraction(double /*ratio*/) const { in setMemoryFraction() function
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DCUDAPluggableAllocator.cpp174 void CUDAPluggableAllocator::setMemoryFraction( in setMemoryFraction() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAMallocAsyncAllocator.cpp453 void setMemoryFraction(double fraction, c10::DeviceIndex device) override { in setMemoryFraction() function
H A DCUDACachingAllocator.h322 inline void setMemoryFraction(double fraction, c10::DeviceIndex device) { in setMemoryFraction() function
H A DCUDACachingAllocator.cpp1563 void setMemoryFraction(double fraction) { in setMemoryFraction() function in c10::cuda::CUDACachingAllocator::Native::DeviceCachingAllocator
3250 void setMemoryFraction(double fraction, c10::DeviceIndex device) override { in setMemoryFraction() function in c10::cuda::CUDACachingAllocator::Native::NativeCachingAllocator