Searched defs:raw_alloc (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/pytorch/c10/xpu/ | ||
H A D | XPUCachingAllocator.cpp | 616 void* raw_alloc(size_t size) { in raw_alloc() function in c10::xpu::XPUCachingAllocator::XPUAllocator |
/aosp_15_r20/external/pytorch/torch/csrc/cuda/ | ||
H A D | CUDAPluggableAllocator.cpp | 130 void* CUDAPluggableAllocator::raw_alloc(size_t nbytes) { in raw_alloc() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator |
/aosp_15_r20/external/pytorch/c10/cuda/ | ||
H A D | CUDAMallocAsyncAllocator.cpp | 832 void* raw_alloc(size_t nbytes) override { in raw_alloc() function |
H A D | CUDACachingAllocator.h | 306 inline void* raw_alloc(size_t nbytes) { in raw_alloc() function |
H A D | CUDACachingAllocator.cpp | 3537 void* raw_alloc(size_t nbytes) override { in raw_alloc() function in c10::cuda::CUDACachingAllocator::Native::NativeCachingAllocator |