| /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/ |
| D | SavedModelBundle.java | 27 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/ |
| D | SavedModelBundleTest.java | 35 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/ |
| D | saved_model_bundle_test.cc | 56 const SavedModelBundle& bundle) { in ValidateAssets()
73 const SavedModelBundle& bundle) { in CheckSavedModelBundle()
105 SavedModelBundle bundle; in TEST_F()
119 SavedModelBundle bundle; in TEST_F()
131 SavedModelBundle bundle; in TEST_F()
147 SavedModelBundle bundle; in TEST_F()
163 SavedModelBundle bundle; in TEST_F()
181 SavedModelBundle bundle; in TEST_F()
193 SavedModelBundle bundle; in TEST_F()
205 SavedModelBundle bundle; in TEST_F()
[all …]
|
| D | loader.h | 49 struct SavedModelBundle : public SavedModelBundleInterface { struct
52 ~SavedModelBundle() override { in ~SavedModelBundle() argument
58 SavedModelBundle() = default;
108 SavedModelBundle* const bundle);
|
| D | loader.cc | 283 SavedModelBundle* const bundle) { in LoadSavedModelInternal()
327 SavedModelBundle* const bundle) { in LoadSavedModel()
461 SavedModelBundle legacy_bundle; in LoadSavedModel()
|
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model_test.cc | 54 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 …]
|
| D | freeze_saved_model.cc | 56 const SavedModelBundle& saved_model_bundle, in GetSignatureDefsInputsAndOutputs()
175 Status FreezeGraphDef(const SavedModelBundle& saved_model_bundle, in FreezeGraphDef()
223 Status FreezeSavedModel(const SavedModelBundle& saved_model_bundle, in FreezeSavedModel()
|
| D | freeze_saved_model.h | 36 Status FreezeSavedModel(const SavedModelBundle& saved_model_bundle,
|
| /external/tensorflow/tensorflow/cc/saved_model/python/ |
| D | loader.clif | 3 class SavedModelBundle:
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.h | 57 ConvertSavedModelV1ToMlir(const SavedModelBundle& saved_model,
|
| D | tf_mlir_translate.cc | 143 tensorflow::SavedModelBundle bundle; in SavedModelV1ToMlirImport()
|
| D | import_model.cc | 2838 static StatusOr<mlir::OwningModuleRef> Convert(const SavedModelBundle& bundle, in Convert()
2846 SavedModelV1Importer(const SavedModelBundle& bundle, in SavedModelV1Importer()
2879 const SavedModelBundle& bundle_;
3186 const SavedModelBundle& saved_model, mlir::MLIRContext* context) { in ConvertSavedModelV1ToMlir()
|
| /external/tensorflow/tensorflow/compiler/mlir/python/ |
| D | mlir.cc | 104 tensorflow::SavedModelBundle bundle; in ExperimentalConvertSavedModelV1ToMlir()
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | README.md | 180 specified. The loaded version of SavedModel is referred to as `SavedModelBundle`
185 SavedModelBundle bundle;
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2217 tensorflow::SavedModelBundle bundle; in TF_LoadSessionFromSavedModel()
|