Searched refs:allocator_array (Results 1 – 1 of 1) sorted by relevance
40 C10_API at::Allocator* allocator_array[at::COMPILE_TIME_MAX_DEVICE_TYPES]; variable46 allocator_array[static_cast<int>(t)] = alloc; in SetAllocator()52 auto* alloc = allocator_array[static_cast<int>(t)]; in GetAllocator()