Searched defs:_set_stream_by_id (Results 1 – 3 of 3) sorted by relevance
103 def _set_stream_by_id(stream_id: int, device_index: int, device_type: int): member in DeviceInterface199 _set_stream_by_id = staticmethod(torch.cuda._set_stream_by_id) # type: ignore[assignment] variable in CudaInterface270 _set_stream_by_id = staticmethod(torch.xpu._set_stream_by_id) # type: ignore[assignment] variable in XpuInterface
327 def _set_stream_by_id(stream_id, device_index, device_type): function
610 def _set_stream_by_id(stream_id, device_index, device_type): function