Searched refs:new_restore_ops (Results 1 – 2 of 2) sorted by relevance
255 self._checkpoint.new_restore_ops(restore_ops)963 new_restore_ops, new_tensor_saveables, new_python_saveables = (968 restore_ops.extend(new_restore_ops)
270 def new_restore_ops(self, new_ops): member in _CheckpointRestoreCoordinator307 new_restore_ops = functional_saver.MultiDeviceSaver(310 for name, restore_op in sorted(new_restore_ops.items()):