Home
last modified time | relevance | path

Searched refs:saved_model_bundle (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc54 SavedModelBundle* saved_model_bundle) { in AddSignatureDefToSavedModelBundle() argument
55 MetaGraphDef* meta_graph_def = &saved_model_bundle->meta_graph_def; in AddSignatureDefToSavedModelBundle()
63 SavedModelBundle* saved_model_bundle) { in InitializeSavedModelBundleSession() argument
65 saved_model_bundle->session.reset(NewSession(session_options)); in InitializeSavedModelBundleSession()
66 TF_RETURN_IF_ERROR(saved_model_bundle->session->Create(graph_def)); in InitializeSavedModelBundleSession()
69 return saved_model_bundle->session->Run( in InitializeSavedModelBundleSession()
79 SavedModelBundle* saved_model_bundle) { in AddGraphDefToSavedModelBundle() argument
80 MetaGraphDef* meta_graph_def = &saved_model_bundle->meta_graph_def; in AddGraphDefToSavedModelBundle()
83 saved_model_bundle); in AddGraphDefToSavedModelBundle()
90 const string& init_node, SavedModelBundle* saved_model_bundle) { in AddGraphDefWithOutputsToSavedModelBundle() argument
[all …]
Dfreeze_saved_model.cc56 const SavedModelBundle& saved_model_bundle, in GetSignatureDefsInputsAndOutputs() argument
58 for (auto& sigdef_elem : saved_model_bundle.meta_graph_def.signature_def()) { in GetSignatureDefsInputsAndOutputs()
175 Status FreezeGraphDef(const SavedModelBundle& saved_model_bundle, in FreezeGraphDef() argument
178 GraphDef graph_def = saved_model_bundle.meta_graph_def.graph_def(); in FreezeGraphDef()
197 saved_model_bundle.session.get(), name_to_node_map, variable_node_names, in FreezeGraphDef()
223 Status FreezeSavedModel(const SavedModelBundle& saved_model_bundle, in FreezeSavedModel() argument
227 GetSignatureDefsInputsAndOutputs(saved_model_bundle, inputs, outputs); in FreezeSavedModel()
229 FreezeGraphDef(saved_model_bundle, *outputs, frozen_graph_def)); in FreezeSavedModel()
Dfreeze_saved_model.h36 Status FreezeSavedModel(const SavedModelBundle& saved_model_bundle,