Searched refs:saveables (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | graph_view.py | 241 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 …]
|
| D | base.py | 311 … saveables = self.trackable._gather_saveables_for_checkpoint() # pylint: disable=protected-access
358 saveable_factory = saveables.get(serialized_tensor.name, None)
|
| D | util.py | 1023 saveables = self._gather_saveable_objects()
1024 v1_saver_lib.Saver(saveables).restore(
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 86 def save_op(self, filename_tensor, saveables): argument
107 for saveable in saveables:
126 def bulk_restore(self, filename_tensor, saveables, preferred_shard, argument
146 for saveable in saveables:
196 def _AddSaveOps(self, filename_tensor, saveables): argument
206 save = self.save_op(filename_tensor, saveables)
259 for shard, (device, saveables) in enumerate(per_device):
265 sharded_saves.append(self._AddSaveOps(sharded_filename, saveables))
296 for shard, (device, saveables) in enumerate(per_device):
300 sharded_saves.append(self._AddSaveOps(sharded_filename, saveables))
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 124 saveables = []
133 saveables.append(
137 return saveables
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | saveable_object_util.py | 306 def _add_saveable(saveables, seen_ops, saveable): argument
321 saveables.append(saveable)
344 saveables = []
350 _add_saveable(saveables, seen_ops, converted_saveable_object)
351 return saveables
|
| D | functional_saver.py | 160 device: _SingleDeviceSaver(saveables)
161 for device, saveables in saveables_by_device.items()}
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | shuffle_dataset_serialization_test.py | 135 saveables = [
139 for saveable in saveables:
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | iterator_ops.py | 230 saveables = [
236 saveables, self._latest_filename, sharded=True)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils.py | 735 …saveables = trackable._gather_saveables_for_checkpoint().values() # pylint: disable=protected-acc…
736 if len(saveables) != 1:
738 saveable = list(saveables)[0]
|