Searched refs:test_allocator_clone (Results 1 – 4 of 4) sorted by relevance
42 TEST(XlaTensorTest, test_allocator_clone) { in TEST() argument47 test_allocator_clone(&allocator); in TEST()
9 test_allocator_clone(c10::GetDefaultCPUAllocator()); in TEST()
5 void test_allocator_clone(c10::Allocator* allocator) { in test_allocator_clone() function
11 test_allocator_clone(c10::cuda::CUDACachingAllocator::get()); in TEST()