Searched refs:GetOrCreateAllocator (Results 1 – 4 of 4) sorted by relevance
45 TestAllocator* allocator = GetOrCreateAllocator(local_client_->platform()); in XLA_TEST_F()82 TestAllocator* allocator = GetOrCreateAllocator(local_client_->platform()); in XLA_TEST_F()
98 /* static */ TestAllocator* LocalClientTestBase::GetOrCreateAllocator( in GetOrCreateAllocator() function in xla::LocalClientTestBase159 run_options.set_allocator(GetOrCreateAllocator(local_client_->platform())); in DefaultExecutableRunOptions()
85 static TestAllocator* GetOrCreateAllocator(se::Platform* platform);
58 shape, GetOrCreateAllocator(local_client_->platform()), in AllocateDeviceBuffer()