Searched refs:GetRecordsAndUnRef (Results 1 – 6 of 6) 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()
86 gtl::InlinedVector<AllocRecord, 4> GetRecordsAndUnRef();
174 gtl::InlinedVector<AllocRecord, 4> TrackingAllocator::GetRecordsAndUnRef() { in GetRecordsAndUnRef() function in tensorflow::TrackingAllocator
358 wrapped_allocator.second->GetRecordsAndUnRef(); in ~OpKernelContext()
87 allocator_pair.second->GetRecordsAndUnRef(); in PersistentMemoryTrackingTest()
187 for (auto& record : alloc.second->GetRecordsAndUnRef()) { in Finalize()