Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAAllocatorConfig.cpp9 namespace c10::cuda::CUDACachingAllocator { namespace
H A DCUDAMallocAsyncAllocator.cpp12 namespace c10::cuda::CUDACachingAllocator::CudaMallocAsync { namespace
H A DCUDACachingAllocator.cpp44 namespace cuda::CUDACachingAllocator { namespace