Searched refs:profiling_allocator (Results 1 – 2 of 2) sorted by relevance
12 thread_local CPUProfilingAllocator* profiling_allocator{nullptr}; variable451 profiling_allocator == nullptr, in WithProfilingAllocatorGuard()453 profiling_allocator = allocator; in WithProfilingAllocatorGuard()454 profiling_allocator->set_plan(plan); in WithProfilingAllocatorGuard()458 profiling_allocator->unset_plan(); in ~WithProfilingAllocatorGuard()459 profiling_allocator = nullptr; in ~WithProfilingAllocatorGuard()463 return profiling_allocator; in GetThreadLocalProfilingAllocator()
131 c10::CPUProfilingAllocator profiling_allocator; in TEST() local134 profiling_allocator_guard(&profiling_allocator, &plan); in TEST()147 profiling_allocator_guard(&profiling_allocator, &plan); in TEST()