Home
last modified time | relevance | path

Searched refs:GetThreadLocalCachingAllocator (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/c10/mobile/
H A DCPUCachingAllocator.cpp95 CPUCachingAllocator* GetThreadLocalCachingAllocator() { in GetThreadLocalCachingAllocator() function
101 : prev_caching_allocator_ptr_(GetThreadLocalCachingAllocator()) { in WithCPUCachingAllocatorGuard()
H A DCPUCachingAllocator.h95 CPUCachingAllocator* GetThreadLocalCachingAllocator();
/aosp_15_r20/external/pytorch/c10/core/
H A DCPUAllocator.cpp87 auto allocator_ptr = GetThreadLocalCachingAllocator(); in deleter()
119 auto allocator_ptr = GetThreadLocalCachingAllocator(); in allocate()