Method __init__

tensorflow/python/training/saver.py:743–745

(self, *args, **kwargs)

  """SaverBuilder with support for custom save and restore.
  """
  def __init__(self, *args, **kwargs):
    self._ckpt_dir_or_file = kwargs.pop("checkpoint_dir", True)
    super(PartialRestoreSaverBuilder, self).__init__(*args, **kwargs)

  def restore_op(self, filename_tensor, saveable, preferred_shard):
    """Create ops to restore 'saveable'.