Searched refs:new_restore_ops (Results 1 – 2 of 2) sorted by relevance
126 def new_restore_ops(self, new_ops): member in _CheckpointRestoreCoordinator161 new_restore_ops = functional_saver.restore_from_saveable_objects(164 restore_ops.extend(new_restore_ops)165 for saveable, restore_op in zip(validated_saveables, new_restore_ops):
211 self._checkpoint.new_restore_ops(restore_ops)