Searched refs:cuda_alloc_visitor (Results 1 – 2 of 2) sorted by relevance
293 SubAllocator::Visitor cuda_alloc_visitor = [](void* ptr, int gpu_id, in RegMemVisitors() local299 cuda_alloc_visitor); in RegMemVisitors()
259 SubAllocator::Visitor cuda_alloc_visitor = [this](void* ptr, int gpu_id, in Init() local265 cuda_alloc_visitor); in Init()