Home
last modified time | relevance | path

Searched refs:saveable (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/python/training/tracking/
Dgraph_view.py245 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_restore
[all …]
Dbase.py331 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 …]
Dutil.py147 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)
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
202 saveable = saveable_factory()
212 saveable = saveable_factory
214 [saveable],
[all …]
/external/tensorflow/tensorflow/python/training/saving/
Dfunctional_saver.py47 for saveable in saveable_objects:
48 if not isinstance(saveable, saveable_object.SaveableObject):
50 "Saver expected a list of SaveableObjects, got %s." % (saveable,))
89 for saveable in self._saveable_objects:
90 for spec in saveable.specs:
121 for saveable in saveable_objects:
124 for spec in saveable.specs:
134 for saveable, restored_tensors in zip(saveable_objects,
136 restore_ops.append(saveable.restore(restored_tensors,
Dsaveable_object_util.py298 def _add_saveable(saveables, seen_ops, saveable): argument
310 if saveable.op in seen_ops:
312 saveable.name)
313 saveables.append(saveable)
314 seen_ops.add(saveable.op)
/external/tensorflow/tensorflow/python/training/
Dsaver.py108 for saveable in saveables:
109 for spec in saveable.specs:
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): argument
175 for spec in saveable.specs:
339 for saveable in saveables:
345 for spec in saveable.specs:
[all …]
Dsaver_test_utils.py53 def saveable(self): member in CheckpointedOp
Dsaver_test.py110 "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})
464 saver_module.Saver({"v2": v2.saveable, "v2too": v2.saveable})
467 saver_module.Saver({"v0": v0, "v2": v2.saveable})
480 save = saver_module.Saver([v0, v1, v2.saveable])
501 save = saver_module.Saver([v0, v1, v2.saveable])
526 save2 = saver_module.Saver([v0_2, v1_2, v2_2.saveable])
[all …]
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dserialization_integration_test.py36 saveable = contrib_iterator_ops.make_saveable_from_iterator(iterator)
37 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
Dshuffle_dataset_serialization_test.py136 for saveable in saveables:
137 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
Ddataset_serialization_test_base.py599 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/ops/
Dmodel_ops.py149 saveable = TreeEnsembleVariableSavable(resource_handle, create_op,
151 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
Dstats_accumulator_ops.py190 def saveable(self): member in StatsAccumulator
194 return {"stats_accumulator", self.saveable}
Dquantile_ops.py156 return {"quantile_accumulator", self.saveable}
/external/tensorflow/tensorflow/contrib/tensor_forest/python/ops/
Dstats_ops.py161 saveable = (
164 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
Dmodel_ops.py167saveable = tree_var._gather_saveables_for_checkpoint()["tree_variable"]( # pylint: disable=protec…
169 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_ops_test.py1093 saveable = fn(
1095 return saveable
1126 saveable = self._create_saveable(opaque_param, rnn_mode, num_units,
1128 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
1129 weights_op, biases_op = saveable.format_converter.opaque_to_tf_canonical(
1130 saveable._variables)
1188 saveable = self._create_saveable(opaque_params[i], rnn_mode, num_units,
1190 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
1191 saveables.append(saveable)
Dcudnn_rnn_test.py540 model.rnn.saveable.format_converter._opaque_to_cu_canonical(
541 model.rnn.saveable._variables))
588 saveable1 = model1.rnn.saveable
591 saveable2 = model1.rnn.saveable
/external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/
Dstats_accumulator_ops_test.py136 hessians_1) = accumulator.saveable.serialize()
178 accumulator.saveable.deserialize(
352 hessians_1) = accumulator.saveable.serialize()
402 deserialize = accumulator.saveable.deserialize(
/external/tensorflow/tensorflow/python/ops/
Dlookup_ops.py1585 saveable = MutableHashTable._Saveable(self, name)
1587 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
1843 saveable = DenseHashTable._Saveable(self, name)
1845 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
Dboosted_trees_ops.py140 def saveable(self): member in QuantileAccumulator
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/
Dcudnn_rnn.py271 def saveable(self): member in _CudnnRNN
/external/u-boot/scripts/
Dspelling.txt1008 savable||saveable
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch.py1009 steps_accumulator.saveable.serialize())