Home
last modified time | relevance | path

Searched refs:test_stream_context (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_cpp_extensions_mtia_backend.py107 def test_stream_context(self): member in TestCppExtensionMTIABackend
H A Dtest_cuda_multigpu.py581 def test_stream_context(self): member in TestCudaMultiGPU
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_cuda.py369 def test_stream_context(): function
388 A, B, is_stream_set, is_stream_reset = test_stream_context()