Searched refs:wrapped_allocator (Results 1 – 1 of 1) sorted by relevance
344 for (auto& wrapped_allocator : tracking_state_->wrapped_allocators) { in ~OpKernelContext() local345 wrapped_allocator.second->GetRecordsAndUnRef(); in ~OpKernelContext()366 TrackingAllocator* wrapped_allocator = in get_allocator() local369 std::make_pair(allocator, wrapped_allocator)); in get_allocator()370 return wrapped_allocator; in get_allocator()