Searched defs:active_blocks (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/c10/xpu/ | ||
H A D | XPUCachingAllocator.cpp | 133 ska::flat_hash_set<Block*> active_blocks; // allocated or in use by a stream member in c10::xpu::XPUCachingAllocator::DeviceCachingAllocator |
/aosp_15_r20/external/pytorch/c10/cuda/ | ||
H A D | CUDACachingAllocator.cpp | 1030 ska::flat_hash_set<Block*> active_blocks; member in c10::cuda::CUDACachingAllocator::Native::DeviceCachingAllocator |