Home
last modified time | relevance | path

Searched refs:saveables (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/training/tracking/
Dgraph_view.py241 saveables = None
243 saveables = cached_attributes.get(name, None)
244 if saveables is not None:
245 for saveable in saveables:
249 saveables = None
252 if saveables is None:
258 saveables = (maybe_saveable,)
266 saveables = tuple(saveable_object_util.saveable_objects_for_op(
268 for saveable in saveables:
270 for saveable in saveables:
[all …]
Dbase.py309 … saveables = self.trackable._gather_saveables_for_checkpoint() # pylint: disable=protected-access
356 saveable_factory = saveables.get(serialized_tensor.name, None)
Dutil.py813 saveables = self._gather_saveable_objects()
814 v1_saver_lib.Saver(saveables).restore(
/external/tensorflow/tensorflow/python/training/
Dsaver.py87 def save_op(self, filename_tensor, saveables): argument
108 for saveable in saveables:
127 def bulk_restore(self, filename_tensor, saveables, preferred_shard, argument
147 for saveable in saveables:
199 def _AddSaveOps(self, filename_tensor, saveables): argument
209 save = self.save_op(filename_tensor, saveables)
262 for shard, (device, saveables) in enumerate(per_device):
268 sharded_saves.append(self._AddSaveOps(sharded_filename, saveables))
299 for shard, (device, saveables) in enumerate(per_device):
303 sharded_saves.append(self._AddSaveOps(sharded_filename, saveables))
[all …]
/external/tensorflow/tensorflow/python/training/saving/
Dsaveable_object_util.py298 def _add_saveable(saveables, seen_ops, saveable): argument
313 saveables.append(saveable)
336 saveables = []
342 _add_saveable(saveables, seen_ops, converted_saveable_object)
343 return saveables
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dshuffle_dataset_serialization_test.py132 saveables = [
136 for saveable in saveables:
/external/tensorflow/tensorflow/python/data/experimental/ops/
Diterator_ops.py187 saveables = [_Saveable(i) for i in iterators]
188 self._checkpoint_saver_hook._saver = _CustomSaver(saveables,
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_ops_test.py1177 saveables = []
1191 saveables.append(saveable)
1196 saveables[i].format_converter.opaque_to_tf_canonical(
1197 saveables[i]._variables))
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Destimator.py1418 saveables = variables._all_saveable_objects()
1424 elif saveables:
1425 saver_for_restore = saver.Saver(saveables, sharded=True)