Searched refs:checkpoint_position (Results 1 – 6 of 6) sorted by relevance
70 def __init__(self, checkpoint_position, shape=None): argument71 self.wrapped_value = checkpoint_position.value_tensors()[VARIABLE_VALUE_KEY]76 self._checkpoint_position = checkpoint_position85 def checkpoint_position(self): member in CheckpointInitialValue780 for checkpoint_position in deferred_dependencies_list:781 if not checkpoint_position.is_simple_variable():787 checkpoint_position = max(791 checkpoint_position=checkpoint_position, shape=shape)872 for checkpoint_position in sorted(876 checkpoint_position.restore(trackable)[all …]
1178 for checkpoint_position in deferred_restorations:1179 checkpoint_position.restore(slot_variable)1217 checkpoint_position=slot_variable_position)
1012 for checkpoint_position in deferred_restorations:1013 checkpoint_position.restore(slot_variable)1051 checkpoint_position=slot_variable_position)
172 self._update_uid = initial_value.checkpoint_position.restore_uid
1772 self._update_uid = initial_value.checkpoint_position.restore_uid
1527 self._update_uid = initial_value.checkpoint_position.restore_uid