Searched refs:caching_allocator (Results 1 – 4 of 4) sorted by relevance
9 c10::CPUCachingAllocator caching_allocator; in TEST() local11 &caching_allocator); in TEST()21 c10::CPUCachingAllocator caching_allocator; in TEST() local25 &caching_allocator); in TEST()34 c10::CPUCachingAllocator caching_allocator; in TEST() local38 &caching_allocator); in TEST()
307 c10::CPUCachingAllocator caching_allocator; in main() local310 caching_allocator_guard.emplace(&caching_allocator); in main()
296 c10::CPUCachingAllocator caching_allocator; in main() local299 caching_allocator_guard.emplace(&caching_allocator); in main()
40 auto& caching_allocator = *::c10::cuda::CUDACachingAllocator::get(); \41 auto temp_storage = caching_allocator.allocate(temp_storage_bytes); \