Home
last modified time | relevance | path

Searched refs:saved_model_bundle (Results 1 – 6 of 6) 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.cc52 const SavedModelBundle& saved_model_bundle, in GetSignatureDefsInputsAndOutputs() argument
54 for (auto& sigdef_elem : saved_model_bundle.meta_graph_def.signature_def()) { in GetSignatureDefsInputsAndOutputs()
171 Status FreezeGraphDef(const SavedModelBundle& saved_model_bundle, in FreezeGraphDef() argument
174 GraphDef graph_def = saved_model_bundle.meta_graph_def.graph_def(); in FreezeGraphDef()
193 saved_model_bundle.session.get(), name_to_node_map, variable_node_names, in FreezeGraphDef()
219 Status FreezeSavedModel(const SavedModelBundle& saved_model_bundle, in FreezeSavedModel() argument
223 GetSignatureDefsInputsAndOutputs(saved_model_bundle, inputs, outputs); in FreezeSavedModel()
225 FreezeGraphDef(saved_model_bundle, *outputs, frozen_graph_def)); in FreezeSavedModel()
Dfreeze_saved_model.h36 Status FreezeSavedModel(const SavedModelBundle& saved_model_bundle,
/external/tensorflow/tensorflow/contrib/session_bundle/
Dbundle_shim.cc133 SavedModelBundle* saved_model_bundle) { in LoadSavedModelFromLegacySessionBundlePath() argument
151 saved_model_bundle); in LoadSavedModelFromLegacySessionBundlePath()
339 SessionBundle& session_bundle, SavedModelBundle* saved_model_bundle) { in ConvertSessionBundleToSavedModelBundle() argument
341 saved_model_bundle->session = std::move(session_bundle.session); in ConvertSessionBundleToSavedModelBundle()
344 saved_model_bundle->meta_graph_def = session_bundle.meta_graph_def; in ConvertSessionBundleToSavedModelBundle()
349 &saved_model_bundle->meta_graph_def); in ConvertSessionBundleToSavedModelBundle()
358 SavedModelBundle* saved_model_bundle, bool* is_session_bundle) { in LoadSessionBundleOrSavedModelBundle() argument
368 saved_model_tags, saved_model_bundle); in LoadSessionBundleOrSavedModelBundle()
379 session_options, run_options, export_dir, saved_model_bundle); in LoadSessionBundleOrSavedModelBundle()
Dbundle_shim_test.cc46 void ValidateHalfPlusTwo(const SavedModelBundle& saved_model_bundle, in ValidateHalfPlusTwo() argument
57 TF_ASSERT_OK(saved_model_bundle.session->Run( in ValidateHalfPlusTwo()
70 SavedModelBundle saved_model_bundle; in LoadAndValidateSavedModelBundle() local
73 session_options, run_options, export_dir, tags, &saved_model_bundle, in LoadAndValidateSavedModelBundle()
76 const MetaGraphDef meta_graph_def = saved_model_bundle.meta_graph_def; in LoadAndValidateSavedModelBundle()
98 ValidateHalfPlusTwo(saved_model_bundle, input_tensor_info.name(), in LoadAndValidateSavedModelBundle()
525 SavedModelBundle saved_model_bundle; in TEST() local
528 tags, &saved_model_bundle)); in TEST()
529 const MetaGraphDef meta_graph_def = saved_model_bundle.meta_graph_def; in TEST()
575 SavedModelBundle saved_model_bundle; in TEST() local
[all …]
Dbundle_shim.h57 SessionBundle& session_bundle, SavedModelBundle* saved_model_bundle);