Home
last modified time | relevance | path

Searched refs:PerDevicePool (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/cuda/
DCachingHostAllocator.cpp62 struct PerDevicePool { struct in at::cuda::__anondde4d04d0111::EventPool
66 std::vector<PerDevicePool> pools_;
/external/pytorch/c10/cuda/
DCUDACachingAllocator.cpp807 struct PerDevicePool { struct in c10::cuda::CUDACachingAllocator::Native::__anon9cb273fa0111::EventPool
811 std::vector<PerDevicePool> pools_;