Home
last modified time | relevance | path

Searched defs:raw_alloc_with_stream (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DCUDAPluggableAllocator.cpp137 void* CUDAPluggableAllocator::raw_alloc_with_stream( in raw_alloc_with_stream() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAMallocAsyncAllocator.cpp843 void* raw_alloc_with_stream(size_t nbytes, cudaStream_t stream) override { in raw_alloc_with_stream() function
H A DCUDACachingAllocator.h310 inline void* raw_alloc_with_stream(size_t nbytes, cudaStream_t stream) { in raw_alloc_with_stream() function
H A DCUDACachingAllocator.cpp3548 void* raw_alloc_with_stream(size_t nbytes, cudaStream_t stream) override { in raw_alloc_with_stream() function in c10::cuda::CUDACachingAllocator::Native::NativeCachingAllocator
/aosp_15_r20/external/pytorch/c10/xpu/
H A DXPUCachingAllocator.cpp626 void* raw_alloc_with_stream(size_t size, XPUStream stream) { in raw_alloc_with_stream() function in c10::xpu::XPUCachingAllocator::XPUAllocator