Home
last modified time | relevance | path

Searched refs:InlineMultiStreamGuard (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/c10/core/impl/
DInlineStreamGuard.h200 class InlineMultiStreamGuard {
205 explicit InlineMultiStreamGuard(ArrayRef<Stream> streams) { in InlineMultiStreamGuard() function
216 InlineMultiStreamGuard(const InlineMultiStreamGuard&) = delete;
217 InlineMultiStreamGuard<T>& operator=(const InlineMultiStreamGuard&) = delete;
221 InlineMultiStreamGuard(InlineMultiStreamGuard&& other) = delete;
222 InlineMultiStreamGuard& operator=(InlineMultiStreamGuard&& other) = delete;
224 ~InlineMultiStreamGuard() noexcept { in ~InlineMultiStreamGuard()
/external/pytorch/c10/test/core/impl/
DInlineStreamGuard_test.cpp179 using MultiTestGuard = InlineMultiStreamGuard<TestGuardImpl>;
181 TEST(InlineMultiStreamGuard, Constructor) { in TEST() argument
/external/pytorch/c10/core/
DStreamGuard.h167 c10::impl::InlineMultiStreamGuard<impl::VirtualGuardImpl> guard_;
/external/pytorch/c10/cuda/
DCUDAGuard.h289 c10::impl::InlineMultiStreamGuard<impl::CUDAGuardImpl> guard_;
/external/pytorch/aten/src/ATen/hip/impl/
DHIPGuardImplMasqueradingAsCUDA.h362 c10::impl::InlineMultiStreamGuard<HIPGuardImplMasqueradingAsCUDA> guard_;