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