Searched refs:GetRecordsAndUnRef (Results 1 – 7 of 7) sorted by relevance
85 auto records = ta->GetRecordsAndUnRef(); in TEST()109 records = ta->GetRecordsAndUnRef(); in TEST()135 auto records = ta->GetRecordsAndUnRef(); in TEST()158 EXPECT_EQ(0, ta->GetRecordsAndUnRef().size()); in TEST()176 EXPECT_EQ(0, ta->GetRecordsAndUnRef().size()); in TEST()
87 gtl::InlinedVector<AllocRecord, 4> GetRecordsAndUnRef();
174 gtl::InlinedVector<AllocRecord, 4> TrackingAllocator::GetRecordsAndUnRef() { in GetRecordsAndUnRef() function in tensorflow::TrackingAllocator
312 wrapped_alloator.second->GetRecordsAndUnRef(); in ~OpKernelContext()
83 allocator_pair.second->GetRecordsAndUnRef(); in PersistentMemoryTrackingTest()
193 allocator_pair.second->GetRecordsAndUnRef(); in UpdateStats()
189 for (auto& record : alloc.second->GetRecordsAndUnRef()) { in Finalize()