Home
last modified time | relevance | path

Searched defs:getBaseAllocation (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DCUDAPluggableAllocator.cpp197 void* CUDAPluggableAllocator::getBaseAllocation(void* ptr, size_t* size) { in getBaseAllocation() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAMallocAsyncAllocator.cpp574 void* getBaseAllocation(void* ptr, size_t* size) override { in getBaseAllocation() function
H A DCUDACachingAllocator.h334 inline void* getBaseAllocation(void* ptr, size_t* size) { in getBaseAllocation() function
H A DCUDACachingAllocator.cpp1505 void* getBaseAllocation(Block* block, size_t* outSize) { in getBaseAllocation() function in c10::cuda::CUDACachingAllocator::Native::DeviceCachingAllocator
3326 void* getBaseAllocation(void* ptr, size_t* outSize) override { in getBaseAllocation() function in c10::cuda::CUDACachingAllocator::Native::NativeCachingAllocator