Searched refs:cpu_caching_alloc (Results 1 – 1 of 1) sorted by relevance
296 C10_API at::Allocator* cpu_caching_alloc = nullptr; variable301 cpu_caching_alloc = alloc; in SetCPUCachingAllocator()307 if (cpu_caching_alloc == nullptr) { in GetCPUCachingAllocator()312 return cpu_caching_alloc; in GetCPUCachingAllocator()