Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCachingHostAllocator.cpp22 class EventPool { class
27 EventPool() : pools_(at::cuda::device_count()) {} in EventPool() function in at::cuda::__anon812ef5cd0111::EventPool
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Devent_pool.cc32 EventPool::EventPool(bool allow_reuse) in EventPool() function in xla::EventPool
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDACachingAllocator.cpp766 class EventPool { class
770 EventPool() : pools_(at::cuda::device_count()) {} in EventPool() function in c10::cuda::CUDACachingAllocator::Native::__anon8687b97a0111::EventPool