Searched defs:roundup_power2_divisions (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/c10/cuda/ | ||
H A D | CUDAAllocatorConfig.h | 62 static std::vector<size_t> roundup_power2_divisions() { in roundup_power2_divisions() function |
H A D | CUDAAllocatorConfig.cpp | 24 size_t CUDAAllocatorConfig::roundup_power2_divisions(size_t size) { in roundup_power2_divisions() function in c10::cuda::CUDACachingAllocator::CUDAAllocatorConfig |
H A D | CUDACachingAllocator.h | 162 std::vector<size_t> roundup_power2_divisions; member |