Searched refs:allocation_count (Results 1 – 8 of 8) sorted by relevance
50 int64 allocation_count_before = allocator_->allocation_count(); in XLA_TEST_F()65 EXPECT_GT(allocator_->allocation_count(), allocation_count_before); in XLA_TEST_F()88 int64 device_allocation_count_before = allocator->allocation_count(d); in XLA_TEST_F()89 int64 allocation_count_before = allocator->allocation_count(); in XLA_TEST_F()99 EXPECT_GT(allocator->allocation_count(), allocation_count_before); in XLA_TEST_F()100 EXPECT_GT(allocator->allocation_count(d), device_allocation_count_before); in XLA_TEST_F()
54 int64 allocation_count() const;55 int64 allocation_count(int device_ordinal) const;
62 int64 TestAllocator::allocation_count() const { in allocation_count() function in xla::TestAllocator67 int64 TestAllocator::allocation_count(int device_ordinal) const { in allocation_count() function in xla::TestAllocator
7909 static intptr_t allocation_count = ALLOC_ALWAYS_SUCCEED; variable7915 if (allocation_count == 0) in duff_allocator()7917 if (allocation_count != ALLOC_ALWAYS_SUCCEED) in duff_allocator()7918 allocation_count--; in duff_allocator()7942 allocation_count = i; in START_TEST()7963 allocation_count = i; in START_TEST()8165 allocation_count = ALLOC_ALWAYS_SUCCEED; in alloc_setup()8189 allocation_count = i; in START_TEST()8254 allocation_count = i; in START_TEST()8284 allocation_count = i; in START_TEST()[all …]
145 if (alloc.allocs == 0 && alloc.allocation_count == allocated) in Dump()172 sample->set_alloc_count(alloc.allocation_count); in Dump()176 dead_callstack_allocations_.emplace_back(it, alloc.allocation_count); in Dump()
295 uint64_t allocation_count = 0; member323 callstack_allocations->allocation_count++; in AddToCallstackAllocations()
37 unsigned allocation_count() const { return allocation_count_; } in allocation_count() function
2938 buffer_pos = utoa(node->allocation_count(), buffer, buffer_pos); in SerializeTraceNode()