Searched refs:TestAllocator (Results 1 – 5 of 5) sorted by relevance
37 /* static */ TestAllocator* LocalClientTestBase::allocator_;39 StatusOr<OwningDeviceMemory> TestAllocator::Allocate(int device_ordinal, in Allocate()52 Status TestAllocator::Deallocate(int device_ordinal, se::DeviceMemoryBase mem) { in Deallocate()62 int64 TestAllocator::allocation_count() const { in allocation_count()67 int64 TestAllocator::allocation_count(int device_ordinal) const { in allocation_count()77 int64 TestAllocator::deallocation_count() const { in deallocation_count()82 int64 TestAllocator::deallocation_count(int device_ordinal) const { in deallocation_count()92 /* static */ TestAllocator* LocalClientTestBase::GetOrCreateAllocator( in GetOrCreateAllocator()98 allocator_ = new TestAllocator( in GetOrCreateAllocator()
42 class TestAllocator : public StreamExecutorMemoryAllocator {44 explicit TestAllocator(se::Platform* platform) in TestAllocator() function80 static TestAllocator* GetOrCreateAllocator(se::Platform* platform);124 static TestAllocator* allocator_;
45 TestAllocator* allocator = GetOrCreateAllocator(local_client_->platform()); in XLA_TEST_F()82 TestAllocator* allocator = GetOrCreateAllocator(local_client_->platform()); in XLA_TEST_F()
698 TestAllocator allocator(wrong_platform); in XLA_TEST_F()
46 class TestAllocator : public DeviceMemoryAllocator { class48 TestAllocator() in TestAllocator() function in xla::__anone4c562340111::TestAllocator52 ~TestAllocator() override { in ~TestAllocator()96 TestAllocator allocator; in TEST()113 TestAllocator allocator; in TEST()155 TestAllocator allocator; in BM_TakeSubTree()