Searched defs:_exchange_device (Results 1 – 2 of 2) sorted by relevance
42 _exchange_device = torch._C._xpu_exchangeDevice variable48 def _exchange_device(device: int) -> int: function
82 _exchange_device = torch._C._cuda_exchangeDevice variable85 def _exchange_device(device: int) -> int: function