Home
last modified time | relevance | path

Searched defs:attachAllocatorTraceTracker (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DCUDAPluggableAllocator.cpp303 void CUDAPluggableAllocator::attachAllocatorTraceTracker( in attachAllocatorTraceTracker() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAMallocAsyncAllocator.cpp642 void attachAllocatorTraceTracker(AllocatorTraceTracker tracker) override { in attachAllocatorTraceTracker() function
H A DCUDACachingAllocator.h413 inline void attachAllocatorTraceTracker(AllocatorTraceTracker tracker) { in attachAllocatorTraceTracker() function
H A DCUDACachingAllocator.cpp1149 void attachAllocatorTraceTracker(AllocatorTraceTracker tracker) { in attachAllocatorTraceTracker() function in c10::cuda::CUDACachingAllocator::Native::DeviceCachingAllocator
3315 void attachAllocatorTraceTracker(AllocatorTraceTracker tracker) override { in attachAllocatorTraceTracker() function in c10::cuda::CUDACachingAllocator::Native::NativeCachingAllocator