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()
63 int64 deallocation_count() const;64 int64 deallocation_count(int device_ordinal) const;
83 int64 TestAllocator::deallocation_count() const { in deallocation_count() function in xla::TestAllocator88 int64 TestAllocator::deallocation_count(int device_ordinal) const { in deallocation_count() function in xla::TestAllocator