Searched refs:saved_model_bundle (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/cc/tools/ |
D | freeze_saved_model_test.cc | 54 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 …]
|
D | freeze_saved_model.cc | 52 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()
|
D | freeze_saved_model.h | 36 Status FreezeSavedModel(const SavedModelBundle& saved_model_bundle,
|
/external/tensorflow/tensorflow/contrib/session_bundle/ |
D | bundle_shim.cc | 133 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()
|
D | bundle_shim_test.cc | 46 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 …]
|
D | bundle_shim.h | 57 SessionBundle& session_bundle, SavedModelBundle* saved_model_bundle);
|