Home
last modified time | relevance | path

Searched refs:SavedModelBundle (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
DSavedModelBundle.java27 public class SavedModelBundle implements AutoCloseable { class
31 public SavedModelBundle load() { in load()
32 return SavedModelBundle.load(exportDir, tags, configProto, runOptions); in load()
94 public static SavedModelBundle load(String exportDir, String... tags) { in load()
147 private SavedModelBundle(Graph graph, Session session, byte[] metaGraphDef) { in SavedModelBundle() method in SavedModelBundle
159 private static SavedModelBundle fromHandle( in fromHandle()
163 return new SavedModelBundle(graph, session, metaGraphDef); in fromHandle()
166 private static native SavedModelBundle load( in load()
/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/
DSavedModelBundleTest.java35 try (SavedModelBundle bundle = SavedModelBundle.load(SAVED_MODEL_PATH, "serve")) { in load()
45 SavedModelBundle bundle = SavedModelBundle.load("__BAD__", "serve"); in loadNonExistentBundle()
56 try (SavedModelBundle bundle = SavedModelBundle.loader(SAVED_MODEL_PATH) in loader()
/external/tensorflow/tensorflow/cc/saved_model/
Dloader_test.cc54 const SavedModelBundle& bundle) { in ValidateAssets()
71 const SavedModelBundle& bundle) { in CheckSavedModelBundle()
104 SavedModelBundle bundle; in TEST_F()
118 SavedModelBundle bundle; in TEST_F()
130 SavedModelBundle bundle; in TEST_F()
146 SavedModelBundle bundle; in TEST_F()
162 SavedModelBundle bundle; in TEST_F()
180 SavedModelBundle bundle; in TEST_F()
192 SavedModelBundle bundle; in TEST_F()
204 SavedModelBundle bundle; in TEST_F()
[all …]
Dloader.h31 struct SavedModelBundle { struct
37 ~SavedModelBundle() { in ~SavedModelBundle() argument
43 SavedModelBundle() = default;
53 SavedModelBundle* const bundle);
Dloader.cc263 SavedModelBundle* const bundle) { in LoadSavedModelInternal()
304 SavedModelBundle* const bundle) { in LoadSavedModel()
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc54 SavedModelBundle* saved_model_bundle) { in AddSignatureDefToSavedModelBundle()
63 SavedModelBundle* saved_model_bundle) { in InitializeSavedModelBundleSession()
79 SavedModelBundle* saved_model_bundle) { in AddGraphDefToSavedModelBundle()
90 const string& init_node, SavedModelBundle* saved_model_bundle) { in AddGraphDefWithOutputsToSavedModelBundle()
122 SavedModelBundle saved_model_bundle; in TestFreezeGraphWithoutDependentVariables()
168 SavedModelBundle saved_model_bundle; in TestFreezeGraphWithDependentVariables()
218 SavedModelBundle saved_model_bundle; in TestFreezeGraphWithAndWithoutDependentVariables()
276 SavedModelBundle saved_model_bundle; in TEST_F()
295 SavedModelBundle saved_model_bundle; in TEST_F()
316 SavedModelBundle saved_model_bundle; in TEST_F()
[all …]
Dfreeze_saved_model.cc52 const SavedModelBundle& saved_model_bundle, in GetSignatureDefsInputsAndOutputs()
171 Status FreezeGraphDef(const SavedModelBundle& saved_model_bundle, in FreezeGraphDef()
219 Status FreezeSavedModel(const SavedModelBundle& saved_model_bundle, in FreezeSavedModel()
Dfreeze_saved_model.h36 Status FreezeSavedModel(const SavedModelBundle& saved_model_bundle,
/external/tensorflow/tensorflow/contrib/session_bundle/
Dbundle_shim.h57 SessionBundle& session_bundle, SavedModelBundle* saved_model_bundle);
68 SavedModelBundle* bundle, bool* is_session_bundle = nullptr);
Dbundle_shim_test.cc46 void ValidateHalfPlusTwo(const SavedModelBundle& saved_model_bundle, in ValidateHalfPlusTwo()
70 SavedModelBundle saved_model_bundle; in LoadAndValidateSavedModelBundle()
525 SavedModelBundle saved_model_bundle; in TEST()
575 SavedModelBundle saved_model_bundle; in TEST()
591 SavedModelBundle saved_model_bundle; in TEST()
Dbundle_shim.cc133 SavedModelBundle* saved_model_bundle) { in LoadSavedModelFromLegacySessionBundlePath()
339 SessionBundle& session_bundle, SavedModelBundle* saved_model_bundle) { in ConvertSessionBundleToSavedModelBundle()
358 SavedModelBundle* saved_model_bundle, bool* is_session_bundle) { in LoadSessionBundleOrSavedModelBundle()
/external/tensorflow/tensorflow/cc/saved_model/python/
Dloader.clif3 class SavedModelBundle:
/external/tensorflow/tensorflow/python/saved_model/
DREADME.md180 specified. The loaded version of SavedModel is referred to as `SavedModelBundle`
185 SavedModelBundle bundle;
/external/tensorflow/tensorflow/c/
Dc_api.cc2629 tensorflow::SavedModelBundle bundle; in TF_LoadSessionFromSavedModel()