Home
last modified time | relevance | path

Searched defs:CUDAPluggableAllocator (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DCUDAPluggableAllocator.h69 struct TORCH_CUDA_CPP_API CUDAPluggableAllocator struct
76 CUDAPluggableAllocator& operator=(CUDAPluggableAllocator& other) = delete; argument
183 } // namespace torch::cuda::CUDAPluggableAllocator argument
H A DCUDAPluggableAllocator.cpp9 namespace torch::cuda::CUDAPluggableAllocator { namespace
45 CUDAPluggableAllocator::CUDAPluggableAllocator( in CUDAPluggableAllocator() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator
50 CUDAPluggableAllocator::CUDAPluggableAllocator(CUDAPluggableAllocator& other) in CUDAPluggableAllocator() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator
/aosp_15_r20/external/pytorch/torch/cuda/
H A Dmemory.py919 class CUDAPluggableAllocator(_CUDAAllocator): class