Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dcuda_stream_test.cpp181 std::unordered_set<cudaStream_t> stream_set{}; in TEST() local
/aosp_15_r20/external/pytorch/c10/xpu/
H A DXPUCachingAllocator.cpp30 using stream_set = ska::flat_hash_set<xpu::XPUStream>; typedef
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDACachingAllocator.cpp136 using stream_set = ska::flat_hash_set<cuda::CUDAStream>; typedef