Searched refs:test_stream_context (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/test/ | ||
H A D | test_cpp_extensions_mtia_backend.py | 107 def test_stream_context(self): member in TestCppExtensionMTIABackend |
H A D | test_cuda_multigpu.py | 581 def test_stream_context(self): member in TestCudaMultiGPU |
/aosp_15_r20/external/pytorch/test/jit/ | ||
H A D | test_cuda.py | 369 def test_stream_context(): function |