Home
last modified time | relevance | path

Searched defs:active_pool (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/cuda/
Dmemory.py1015 def active_pool() -> Optional[_MemPool]: member in MemPoolContext
/external/pytorch/c10/cuda/
DCUDACachingAllocator.cpp2657 auto active_pool = MemPoolContext::getActiveMemPool(); in alloc_block() local