Searched defs:releasePool (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/cuda/ | ||
| H A D | CUDAPluggableAllocator.cpp | 276 void CUDAPluggableAllocator::releasePool( in releasePool() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator |
| /aosp_15_r20/external/pytorch/c10/cuda/ | ||
| H A D | CUDAMallocAsyncAllocator.cpp | 817 void releasePool(c10::DeviceIndex device, MempoolId_t mempool_id) override { in releasePool() function |
| H A D | CUDACachingAllocator.h | 417 inline void releasePool(c10::DeviceIndex device, MempoolId_t mempool_id) { in releasePool() function |
| H A D | CUDACachingAllocator.cpp | 2061 void releasePool(MempoolId_t mempool_id) { in releasePool() function in c10::cuda::CUDACachingAllocator::Native::DeviceCachingAllocator |