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