Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/training/saving/
Dfunctional_saver.py51 for saveable in saveable_objects:
52 if not isinstance(saveable, saveable_object.SaveableObject):
54 "Expected a list of SaveableObjects, got %s." % (saveable,))
71 for saveable in self._saveable_objects:
72 for spec in saveable.specs:
99 for saveable in self._saveable_objects:
102 for spec in saveable.specs:
113 for saveable, restored_tensors in zip(self._saveable_objects,
115 restore_ops[saveable.name] = saveable.restore(
156 for saveable in saveable_objects:
[all …]
Dsaveable_object_util.py319 def _add_saveable(saveables, seen_ops, saveable): argument
331 if saveable.op is not None and saveable.op in seen_ops:
333 saveable.name)
334 saveables.append(saveable)
335 seen_ops.add(saveable.op)
406 for saveable in saveables:
407 for spec in saveable.specs:
413 if any(isinstance(saveable, trackable.PythonStateSaveable)
414 for saveable in saveables):
420 if any(isinstance(saveable, trackable.NoRestoreSaveable)
[all …]
Dfunctional_saver_test.py208 saveable = DummyHook(name="dummy")
210 saver = functional_saver.MultiDeviceSaver([saveable])
/external/tensorflow/tensorflow/python/training/tracking/
Dgraph_view.py269 for saveable in saveables:
270 if attribute.checkpoint_key not in saveable.name:
294 for saveable in saveables:
295 saveable.full_name = full_name
296 for saveable in saveables:
297 if attribute.checkpoint_key not in saveable.name:
301 % (trackable, name, saveable.name,
307 for saveable in saveables:
309 optional_restore = saveable.optional_restore
311 optional_restore = optional_restore and saveable.optional_restore
[all …]
Dbase.py394 saveable = None
404 saveable, = saveable_list
409 saveable = None
410 if saveable is not None:
413 if serialized_tensor.checkpoint_key not in saveable.name:
414 saveable = None
416 if saveable is None:
430 saveable = saveable_factory(name=serialized_tensor.checkpoint_key)
432 saveable = saveable_factory
435 {})[serialized_tensor.name] = [saveable]
[all …]
Dutil.py290 for saveable in python_saveables:
295 spec_names = [spec.name for spec in saveable.specs]
296 saveable.python_restore([reader.get_tensor(name) for name in spec_names])
302 validated_names = set(saveable.name for saveable in validated_saveables)
351 saveable = saveable_factory()
361 saveable = saveable_factory
363 [saveable], convert_variable_to_tensor=False)
375 for saveable in self.globally_named_object_attributes(trackable):
378 for spec in saveable.specs:
393 self.unused_attributes.setdefault(trackable, []).append(saveable.name)
[all …]
/external/tensorflow/tensorflow/python/training/
Dsaver.py106 for saveable in saveables:
107 for spec in saveable.specs:
145 for saveable in saveables:
146 if saveable.device:
147 device = saveable_object_util.set_cpu0(saveable.device)
152 self.restore_op(filename_tensor, saveable, preferred_shard))
156 def restore_op(self, filename_tensor, saveable, preferred_shard): argument
173 for spec in saveable.specs:
344 for saveable in saveables:
350 for spec in saveable.specs:
[all …]
Dsaver_test_utils.py53 def saveable(self): member in CheckpointedOp
Dsaver_test.py106 "v2": v2.saveable
126 save = saver_module.Saver({"v0": v0, "v1": v1, "v2": v2.saveable})
154 save2 = saver_module.Saver({"v0": v0_2, "v1": v1_2, "v2": v2_2.saveable})
321 "v2": v2.saveable
356 save = saver_module.Saver({"v0": v0, "v1": v1, "v2": v2.saveable})
463 saver_module.Saver({"v2": v2.saveable, "v2too": v2.saveable})
466 saver_module.Saver({"v0": v0, "v2": v2.saveable})
479 save = saver_module.Saver([v0, v1, v2.saveable])
500 save = saver_module.Saver([v0, v1, v2.saveable])
525 save2 = saver_module.Saver([v0_2, v1_2, v2_2.saveable])
[all …]
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dshuffle_dataset_serialization_test.py98 for saveable in saveables:
99 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
Dserialization_integration_test.py40 saveable = contrib_iterator_ops.make_saveable_from_iterator(iterator)
41 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
Ddataset_serialization_test_base.py432 saveable = contrib_iterator_ops.make_saveable_from_iterator(iterator)
433 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_utils.py813 saveable = list(saveables)[0]
821 self._saveable = saveable
831 self._saveable = saveable()
/external/tensorflow/tensorflow/python/ops/
Dlookup_ops.py1787 saveable = MutableHashTable._Saveable(self, name)
1789 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
2069 saveable = DenseHashTable._Saveable(self, name)
2071 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
Dboosted_trees_ops.py146 def saveable(self): member in QuantileAccumulator
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.__internal__.-func-graph.pbtxt55 name: "saveable"
/external/tensorflow/tensorflow/python/saved_model/
Dsave.py355 if not concrete_function.graph.saveable:
/external/tensorflow/tensorflow/python/framework/
Dfunc_graph.py792 def saveable(self): member in FuncGraph
/external/tensorflow/tensorflow/python/eager/
Dfunction.py1892 if default_graph.building_function and not self._func_graph.saveable:
/external/kotlinx.coroutines/benchmarks/src/jmh/resources/
Dospd.txt.gz