Home
last modified time | relevance | path

Searched defs:expected_live_allocations (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDACachingAllocator.h229 const std::unordered_set<void*>& expected_live_allocations) { in checkPoolLiveAllocations()
404 const std::unordered_set<void*>& expected_live_allocations) { in checkPoolLiveAllocations()
H A DCUDACachingAllocator.cpp1118 const std::unordered_set<void*>& expected_live_allocations) { in checkPoolLiveAllocations()
3304 const std::unordered_set<void*>& expected_live_allocations) override { in checkPoolLiveAllocations()
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DModule.cpp1318 const py::set& expected_live_allocations) { in registerCudaPluggableAllocator()