Lines Matching refs:saveables
205 def save_op(self, filename_tensor, saveables):
206 """Create an Op to save 'saveables'.
213 saveables: A list of BaseSaverBuilder.SaveableObject objects.
226 for saveable in saveables:
245 def bulk_restore(self, filename_tensor, saveables, preferred_shard,
247 """Restore all tensors contained in saveables.
250 Subclasses may override to load multiple saveables in a single call.
254 saveables: List of BaseSaverBuilder.SaveableObject objects.
265 for saveable in saveables:
316 def _AddSaveOps(self, filename_tensor, saveables):
321 saveables: A list of SaveableObject objects.
326 save = self.save_op(filename_tensor, saveables)
379 for shard, (device, saveables) in enumerate(per_device):
385 sharded_saves.append(self._AddSaveOps(sharded_filename, saveables))
416 for shard, (device, saveables) in enumerate(per_device):
420 sharded_saves.append(self._AddSaveOps(sharded_filename, saveables))
428 saveables,
433 """Add operations to restore saveables.
437 saveables: A list of SaveableObject objects.
448 all_tensors = self.bulk_restore(filename_tensor, saveables, preferred_shard,
457 for saveable in saveables:
493 for shard, (device, saveables) in enumerate(per_device):
498 saveables,
509 def _GroupByDevices(self, saveables):
517 saveables: A list of BaseSaverBuilder.SaveableObject objects.
527 for saveable in saveables:
552 ValueError: If at least two saveables share the same name.
629 saveables = []
638 self._AddSaveable(saveables, seen_ops, op)
663 self._AddSaveable(saveables, seen_ops, saveable)
684 self._AddSaveable(saveables, seen_ops, saveable)
685 return saveables
687 def _AddSaveable(self, saveables, seen_ops, saveable):
688 """Adds the saveable to the saveables list.
691 saveables: List to append the SaveableObject to.
692 seen_ops: Set of the ops of the saveables already processed. Used to
702 saveables.append(saveable)
774 saveables = self._ValidateAndSliceInputs(names_to_saveables)
779 [saveable.op for saveable in saveables]) as name:
785 per_device = self._GroupByDevices(saveables)
793 save_tensor = self._AddSaveOps(filename_tensor, saveables)
795 restore_op = self._AddRestoreOps(filename_tensor, saveables,
832 """SaverBuilder with support for bulk restoring multiple saveables."""
834 def bulk_restore(self, filename_tensor, saveables, preferred_shard,
840 for saveable in saveables:
