Searched refs:get_device_module (Results 1 – 4 of 4) sorted by relevance
84 default_stream = torch.get_device_module(device).current_stream()
708 get_device_module
2565 def get_device_module(device: _Optional[_Union[torch.device, str]] = None): function
298 torch.get_device_module,