Searched refs:_rebuild_device_tensor_from_numpy (Results 1 – 3 of 3) sorted by relevance
333 def _rebuild_device_tensor_from_numpy(data, dtype, device, requires_grad): function341 _rebuild_xla_tensor = _rebuild_device_tensor_from_numpy
298 torch._utils._rebuild_device_tensor_from_numpy,
554 rebuild_func is torch._utils._rebuild_device_tensor_from_numpy