Searched defs:maybe_exchange_device (Results 1 – 3 of 3) sorted by relevance
75 def maybe_exchange_device(device: int) -> int: member in DeviceInterface204 … maybe_exchange_device = staticmethod(torch.cuda._maybe_exchange_device) # type: ignore[arg-type] variable in CudaInterface275 maybe_exchange_device = staticmethod(torch.xpu._maybe_exchange_device) # type: ignore[arg-type] variable in XpuInterface
83 def maybe_exchange_device(device: int) -> int: member in DeviceInterface
205 c10::DeviceIndex maybe_exchange_device(c10::DeviceIndex to_device) { in maybe_exchange_device() function