Searched defs:get_raw_stream (Results 1 – 3 of 3) sorted by relevance
107 def get_raw_stream(device_idx: int) -> int: member in DeviceInterface202 get_raw_stream = staticmethod(get_cuda_stream) # type: ignore[assignment, arg-type] variable in CudaInterface273 get_raw_stream = staticmethod(get_xpu_stream) # type: ignore[assignment, arg-type] variable in XpuInterface
105 def get_raw_stream(device_index: int): member in DeviceInterface
649 from torch._C import _cuda_getCurrentRawStream as get_raw_stream unknown