Searched refs:checkpoint_position (Results 1 – 8 of 8) sorted by relevance
76 def __init__(self, checkpoint_position, shape=None): argument77 self.wrapped_value = checkpoint_position.value_tensors()[83 self._checkpoint_position = checkpoint_position92 def checkpoint_position(self): member in CheckpointInitialValue659 for checkpoint_position in deferred_dependencies_list:660 if not checkpoint_position.is_simple_variable():666 checkpoint_position = max(670 checkpoint_position=checkpoint_position, shape=shape)752 for checkpoint_position in sorted(756 checkpoint_position.restore(trackable)[all …]
389 checkpoint_position=slot_variable_position)454 for checkpoint_position in sorted(458 checkpoint_position.restore(v)470 for checkpoint_position in deferred_restorations:471 checkpoint_position.restore(slot_variable)
858 for checkpoint_position in deferred_restorations:859 checkpoint_position.restore(slot_variable)897 checkpoint_position=slot_variable_position)
1155 for checkpoint_position in deferred_restorations:1156 checkpoint_position.restore(slot_variable)1194 checkpoint_position=slot_variable_position)
134 self._update_uid = initial_value.checkpoint_position.restore_uid
118 self._update_uid = initial_value.checkpoint_position.restore_uid
510 self._update_uid = initial_value.checkpoint_position.restore_uid
1525 self._update_uid = initial_value.checkpoint_position.restore_uid