/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...] |