| /external/tensorflow/tensorflow/python/training/saving/ |
| functional_saver.py | 47 for saveable in saveable_objects:
48 if not isinstance(saveable, saveable_object.SaveableObject):
50 "Saver expected a list of SaveableObjects, got %s." % (saveable,))
77 """Save the saveable objects to a checkpoint with `file_prefix`.
89 for saveable in self._saveable_objects:
90 for spec in saveable.specs:
100 """Restore the saveable objects from a checkpoint with `file_prefix`.
121 for saveable in saveable_objects:
124 for spec in saveable.specs:
134 for saveable, restored_tensors in zip(saveable_objects
[all...] |
| saveable_object_util.py | 101 "Saveable is neither a resource variable nor a read operation."
298 def _add_saveable(saveables, seen_ops, saveable):
299 """Adds the saveable to the saveables list.
304 check that each saveable is only saved once.
305 saveable: The saveable.
308 ValueError: If the saveable has already been processed.
310 if saveable.op in seen_ops:
311 raise ValueError("The same saveable will be restored with two names: %s" %
312 saveable.name
[all...] |
| /external/tensorflow/tensorflow/python/training/tracking/ |
| graph_view.py | 245 for saveable in saveables:
246 if attribute.checkpoint_key not in saveable.name:
268 for saveable in saveables:
269 saveable.full_name = full_name
270 for saveable in saveables:
271 if attribute.checkpoint_key not in saveable.name:
275 % (trackable, name, saveable.name,
281 for saveable in saveables:
283 optional_restore = saveable.optional_restore
285 optional_restore = optional_restore and saveable.optional_restor
[all...] |
| base.py | 331 saveable = None
340 saveable, = saveable_list
345 saveable = None
346 if saveable is not None:
349 if serialized_tensor.checkpoint_key not in saveable.name:
350 saveable = None
353 if saveable is None:
367 saveable = saveable_factory(name=serialized_tensor.checkpoint_key)
369 saveable = saveable_factory
372 self.trackable, {})[serialized_tensor.name] = [saveable]
[all...] |
| util.py | 147 for saveable in python_saveables:
148 spec_names = [spec.name for spec in saveable.specs]
149 saveable.python_restore(
156 validated_names = set(saveable.name for saveable in validated_saveables)
159 ("Saveable keys changed when validating. Got back %s, was "
165 for saveable, restore_op in zip(validated_saveables, new_restore_ops):
166 assert saveable.name not in self.restore_ops_by_name
167 self.restore_ops_by_name[saveable.name] = restore_op
198 # This saveable object factory does not have a default name= argument
[all...] |
| /external/tensorflow/tensorflow/python/training/ |
| saver.py | 108 for saveable in saveables:
109 for spec in saveable.specs:
131 By default, this issues separate calls to `restore_op` for each saveable.
141 A list of Tensors resulting from reading 'saveable' from
147 for saveable in saveables:
148 if saveable.device:
149 device = saveable_object_util.set_cpu0(saveable.device)
154 self.restore_op(filename_tensor, saveable, preferred_shard))
158 def restore_op(self, filename_tensor, saveable, preferred_shard):
159 """Create ops to restore 'saveable'
[all...] |
| saver_test_utils.py | 53 def saveable(self): member in class:CheckpointedOp
76 """A custom saveable for CheckpointedOp."""
|
| saver_test.py | 110 "v2": v2.saveable
130 save = saver_module.Saver({"v0": v0, "v1": v1, "v2": v2.saveable})
158 save2 = saver_module.Saver({"v0": v0_2, "v1": v1_2, "v2": v2_2.saveable})
324 "v2": v2.saveable},
358 save = saver_module.Saver({"v0": v0, "v1": v1, "v2": v2.saveable})
458 ValueError, "The same saveable will be restored with two names: v0"):
463 ValueError, "The same saveable will be restored with two names: v2"):
464 saver_module.Saver({"v2": v2.saveable, "v2too": v2.saveable})
467 saver_module.Saver({"v0": v0, "v2": v2.saveable})
[all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| model_ops.py | 149 saveable = TreeEnsembleVariableSavable(resource_handle, create_op,
151 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
|
| stats_accumulator_ops.py | 190 def saveable(self): member in class:StatsAccumulator
194 return {"stats_accumulator", self.saveable}
|
| quantile_ops.py | 156 return {"quantile_accumulator", self.saveable}
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ops/ |
| stats_ops.py | 161 saveable = (
164 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
|
| model_ops.py | 167 saveable = tree_var._gather_saveables_for_checkpoint()["tree_variable"]( # pylint: disable=protected-access
169 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| serialization_integration_test.py | 36 saveable = contrib_iterator_ops.make_saveable_from_iterator(iterator)
37 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
|
| shuffle_dataset_serialization_test.py | 136 for saveable in saveables:
137 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
|
| dataset_serialization_test_base.py | 62 # (deprecated) saveable `SparseTensorSliceDataset`, once the API
411 """Attempts to save a non-saveable iterator.
599 saveable = contrib_iterator_ops.make_saveable_from_iterator(iterator)
600 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
616 saveable = contrib_iterator_ops.make_saveable_from_iterator(iterator)
617 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| stats_accumulator_ops_test.py | 136 hessians_1) = accumulator.saveable.serialize()
178 accumulator.saveable.deserialize(
352 hessians_1) = accumulator.saveable.serialize()
402 deserialize = accumulator.saveable.deserialize(
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| cudnn_rnn_ops_test.py | [all...] |
| cudnn_rnn_test.py | 408 "test-saveable-device-assignment")
540 model.rnn.saveable.format_converter._opaque_to_cu_canonical(
541 model.rnn.saveable._variables))
588 saveable1 = model1.rnn.saveable
591 saveable2 = model1.rnn.saveable
688 "Cudnn saveable already created"):
[all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/ |
| cudnn_rnn.py | 271 def saveable(self): member in class:_CudnnRNN
365 # Create saveable in the outer scope of the cudnn subgraph, such that
516 """Create custom saveable for the Cudnn layer.
523 RuntimeError: if any custom saveable is already created for this layer.
526 raise RuntimeError("Cudnn saveable already created.")
|
| /external/tensorflow/tensorflow/python/ops/ |
| boosted_trees_ops.py | 140 def saveable(self): member in class:QuantileAccumulator
|
| lookup_ops.py | [all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| gbdt_batch.py | [all...] |
