Searched defs:get_cuda_stream (Results 1 – 2 of 2) sorted by relevance
11 from torch._C import _cuda_getCurrentRawStream as get_cuda_stream unknown13 get_cuda_stream = None variable
412 from torch._C import _cuda_getCurrentRawStream as get_cuda_stream unknown