Searched defs:from_device (Results 1 – 8 of 8) sorted by relevance
41 Device* from_device; in RecvFromPeer() local99 Device* from_device, DeviceContext* from_device_ctx, in PostToPeer()
72 const string& key, Device* from_device, in PostToPeer()
541 auto from_device = GetInstructionDevice(*channel.send); in DumpCollectedStats() local
258 def from_device( member in JsonKeyStore
88 def from_device(cls, device: torch.device) -> "_FSDPDeviceHandle": member in _FSDPDeviceHandle
143 pub fn from_device(device: &dyn BusDevice) -> Self { in from_device() method
1084 def _recv(self, dtype, tensor_name, from_device): argument
1589 auto from_device = DeviceFromContext(from); in CanEnablePeerAccess() local