Home
last modified time | relevance | path

Searched refs:getBaseAllocation (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/c10/cuda/
DCUDACachingAllocator.h210 virtual void* getBaseAllocation(void* ptr, size_t* size) = 0;
334 inline void* getBaseAllocation(void* ptr, size_t* size) { in getBaseAllocation() function
335 return get()->getBaseAllocation(ptr, size); in getBaseAllocation()
DCUDAMallocAsyncAllocator.cpp574 void* getBaseAllocation(void* ptr, size_t* size) override { in getBaseAllocation() function
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
3331 return device_allocator[block->device]->getBaseAllocation(block, outSize); in getBaseAllocation()
/external/pytorch/torch/csrc/cuda/
DCUDAPluggableAllocator.h114 void* getBaseAllocation(void* ptr, size_t* size) override;
DCUDAPluggableAllocator.cpp197 void* CUDAPluggableAllocator::getBaseAllocation(void* ptr, size_t* size) { in getBaseAllocation() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator