Searched defs:attachOutOfMemoryObserver (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/cuda/ | ||
| H A D | CUDAPluggableAllocator.cpp | 295 void CUDAPluggableAllocator::attachOutOfMemoryObserver( in attachOutOfMemoryObserver() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator |
| /aosp_15_r20/external/pytorch/c10/cuda/ | ||
| H A D | CUDAMallocAsyncAllocator.cpp | 635 void attachOutOfMemoryObserver(OutOfMemoryObserver observer) override { in attachOutOfMemoryObserver() function |
| H A D | CUDACachingAllocator.h | 409 inline void attachOutOfMemoryObserver(OutOfMemoryObserver observer) { in attachOutOfMemoryObserver() function |
| H A D | CUDACachingAllocator.cpp | 1145 void attachOutOfMemoryObserver(OutOfMemoryObserver observer) { in attachOutOfMemoryObserver() function in c10::cuda::CUDACachingAllocator::Native::DeviceCachingAllocator |