MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / restore

Method restore

tensorflow/python/distribute/values.py:968–973  ·  view source on GitHub ↗

Restore the same value into all variables.

(self, restored_tensors, restored_shapes)

Source from the content-addressed store, hash-verified

966 super(_MirroredSaveable, self).__init__(primary_variable, "", name)
967
968 def restore(self, restored_tensors, restored_shapes):
969 """Restore the same value into all variables."""
970 tensor, = restored_tensors
971 return control_flow_ops.group(tuple(
972 _assign_on_device(v.device, v, tensor)
973 for v in self._mirrored_variable.values))
974
975
976class MirroredVariable(DistributedVariable, Mirrored):

Calls 3

tupleFunction · 0.85
_assign_on_deviceFunction · 0.85
groupMethod · 0.45