Searched refs:_assign_on_device (Results 1 – 1 of 1) sorted by relevance
408 def _assign_on_device(device, variable, tensor): function643 _assign_on_device(v.device, v, tensor)1254 _assign_on_device(v.device, v, tensor) for v in self._values))