Searched defs:attachAllocatorTraceTracker (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/csrc/cuda/ | ||
| D | CUDAPluggableAllocator.cpp | 303 void CUDAPluggableAllocator::attachAllocatorTraceTracker( in attachAllocatorTraceTracker() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator |
| /external/pytorch/c10/cuda/ | ||
| D | CUDAMallocAsyncAllocator.cpp | 642 void attachAllocatorTraceTracker(AllocatorTraceTracker tracker) override { in attachAllocatorTraceTracker() function |
| D | CUDACachingAllocator.h | 413 inline void attachAllocatorTraceTracker(AllocatorTraceTracker tracker) { in attachAllocatorTraceTracker() function |
| D | CUDACachingAllocator.cpp | 1149 void attachAllocatorTraceTracker(AllocatorTraceTracker tracker) { in attachAllocatorTraceTracker() function in c10::cuda::CUDACachingAllocator::Native::DeviceCachingAllocator |