Searched defs:beginAllocateToPool (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/cuda/ | ||
| H A D | CUDAPluggableAllocator.cpp | 259 void CUDAPluggableAllocator::beginAllocateToPool( in beginAllocateToPool() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator |
| /aosp_15_r20/external/pytorch/c10/cuda/ | ||
| H A D | CUDAMallocAsyncAllocator.cpp | 769 void beginAllocateToPool( in beginAllocateToPool() function |
| H A D | CUDACachingAllocator.h | 372 inline void beginAllocateToPool( in beginAllocateToPool() function |
| H A D | CUDACachingAllocator.cpp | 2021 void beginAllocateToPool( in beginAllocateToPool() function in c10::cuda::CUDACachingAllocator::Native::DeviceCachingAllocator |