Searched refs:saveables (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 85 def save_op(self, filename_tensor, saveables): argument
106 for saveable in saveables:
125 def bulk_restore(self, filename_tensor, saveables, preferred_shard, argument
145 for saveable in saveables:
195 def _AddSaveOps(self, filename_tensor, saveables): argument
205 save = self.save_op(filename_tensor, saveables)
267 for shard, (device, saveables) in enumerate(per_device):
273 sharded_saves.append(self._AddSaveOps(sharded_filename, saveables))
304 for shard, (device, saveables) in enumerate(per_device):
308 sharded_saves.append(self._AddSaveOps(sharded_filename, saveables))
[all …]
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | graph_view.py | 265 saveables = None
267 saveables = cached_attributes.get(name, None)
268 if saveables is not None:
269 for saveable in saveables:
273 saveables = None
276 if saveables is None:
284 saveables = (maybe_saveable,)
292 saveables = tuple(saveable_object_util.saveable_objects_for_op(
294 for saveable in saveables:
296 for saveable in saveables:
[all …]
|
| D | base.py | 372 … saveables = self.trackable._gather_saveables_for_checkpoint() # pylint: disable=protected-access
419 saveable_factory = saveables.get(serialized_tensor.name, None)
|
| D | util.py | 1048 saveables = self._gather_saveable_objects()
1049 v1_saver_lib.Saver(saveables).restore(
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | saveable_object_util.py | 319 def _add_saveable(saveables, seen_ops, saveable): argument
334 saveables.append(saveable)
357 saveables = []
363 _add_saveable(saveables, seen_ops, converted_saveable_object)
364 return saveables
394 saveables = []
402 saveables[:] = maybe_saveable
406 for saveable in saveables:
414 for saveable in saveables):
421 for saveable in saveables):
[all …]
|
| D | functional_saver.py | 174 device: _SingleDeviceSaver(saveables)
175 for device, saveables in saveables_by_device.items()}
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | shuffle_dataset_serialization_test.py | 94 saveables = [
98 for saveable in saveables:
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils.py | 804 …saveables = trackable._gather_saveables_for_checkpoint().values() # pylint: disable=protected-acc…
807 if not saveables:
812 elif len(saveables) == 1:
813 saveable = list(saveables)[0]
844 (trackable, len(saveables)))
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | iterator_ops.py | 234 saveables = [
240 saveables, self._latest_filename, sharded=True)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 541 saveables = []
550 saveables.append(
554 return saveables
|