Searched refs:checkpoint_position (Results 1 – 8 of 8) sorted by relevance
69 def __init__(self, checkpoint_position): argument70 self._checkpoint_position = checkpoint_position73 def checkpoint_position(self): member in CheckpointInitialValueCallable102 def __init__(self, checkpoint_position, shape=None, shard_info=None): argument112 self.wrapped_value = checkpoint_position.value_tensors(118 self._checkpoint_position = checkpoint_position127 def checkpoint_position(self): member in CheckpointInitialValue846 for checkpoint_position in deferred_dependencies_list:847 if not checkpoint_position.is_simple_variable():853 checkpoint_position = max([all …]
161 def _restore_from_checkpoint_position(self, checkpoint_position): argument163 checkpoint_position)165 def _single_restoration_from_checkpoint_position(self, checkpoint_position, argument168 checkpoint_position, visit_queue)
1177 for checkpoint_position in deferred_restorations:1178 checkpoint_position.restore(slot_variable)1216 checkpoint_position=slot_variable_position)
1335 for checkpoint_position in deferred_restorations:1336 checkpoint_position.restore(slot_variable)1380 checkpoint_position=slot_variable_position)
2082 "initial_value"].checkpoint_position.restore_uid2087 "initial_value"].checkpoint_position.restore_uid
1766 self._update_uid = initial_value.checkpoint_position.restore_uid1800 self._update_uid = initial_value.checkpoint_position.restore_uid
1718 self._update_uid = initial_value.checkpoint_position.restore_uid
251 self._update_uid = initial_value.checkpoint_position.restore_uid