Home
last modified time | relevance | path

Searched refs:trackable_object (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/cc/saved_model/
Dbundle_v2.cc154 const TrackableObjectGraph::TrackableObject* trackable_object, in RecurseObjectsToRestore() argument
161 (trackable_object->attributes_size() > 0 || in RecurseObjectsToRestore()
162 trackable_object->slot_variables_size() > 0)) { in RecurseObjectsToRestore()
164 callback(saved_object_node_id, *trackable_object), "Unable to restore ", in RecurseObjectsToRestore()
168 for (const auto& trackable_child_ref : trackable_object->children()) { in RecurseObjectsToRestore()
Dbundle_v2_test.cc38 const TrackableObjectGraph::TrackableObject& trackable_object) in RestoreVarsAndVerify()
40 for (const auto& attr : trackable_object.attributes()) { in RestoreVarsAndVerify()
Dbundle_v2.h74 const TrackableObjectGraph::TrackableObject* trackable_object,
/external/tensorflow/tensorflow/python/training/tracking/
Dutil.py765 for trackable_object in self._graph_view.list_objects():
768 if (isinstance(trackable_object,
770 not trackable_object._checkpoint_dependencies):
772 self._checkpoint.all_python_objects.add(trackable_object)
788 for trackable_object in self._graph_view.list_objects():
789 self._checkpoint.all_python_objects.add(trackable_object)
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_cli.py330 trackable_object = load.load(saved_model_dir)
334 save._AugmentedGraphView(trackable_object) # pylint: disable=protected-access
335 .list_functions(trackable_object))
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc2339 const TrackableObjectGraph::TrackableObject& trackable_object, in FindSerializedTensorInTrackable() argument
2341 for (const auto& maybe_serialized_tensor : trackable_object.attributes()) { in FindSerializedTensorInTrackable()
2587 const TrackableObjectGraph::TrackableObject& trackable_object) { in CreateSavedModelIR() argument
2589 std::make_pair(saved_node_id, &trackable_object)); in CreateSavedModelIR()
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer.py294 def _add_trackable(self, trackable_object, trainable): argument
306 handler = base_layer_utils.TrackableWeightHandler(trackable_object)