Searched refs:deallocation_count (Results 1 – 3 of 3) sorted by relevance
68 int64 deallocation_count_before = allocator_->deallocation_count(); in XLA_TEST_F()70 EXPECT_EQ(deallocation_count_before + 1, allocator_->deallocation_count()); in XLA_TEST_F()
58 int64 deallocation_count() const;59 int64 deallocation_count(int device_ordinal) const;
77 int64 TestAllocator::deallocation_count() const { in deallocation_count() function in xla::TestAllocator82 int64 TestAllocator::deallocation_count(int device_ordinal) const { in deallocation_count() function in xla::TestAllocator