Searched refs:_use_cuda_memory_pool_manager (Results 1 – 2 of 2) sorted by relevance
530 def _use_cuda_memory_pool_manager( function641 ), disable_conv_cache_emptying(), clear_cublas_manager(), _use_cuda_memory_pool_manager(1591 ), _use_cuda_memory_pool_manager(
4241 def _use_cuda_memory_pool_manager(device, mem_pool): function4269 with _use_cuda_memory_pool_manager(device, pool):