Home
last modified time | relevance | path

Searched refs:export_dir (Results 1 – 25 of 56) sorted by relevance

123

/external/tensorflow/tensorflow/cc/saved_model/
Dsaved_model_bundle_test.cc65 void ValidateAssets(const string& export_dir, in ValidateAssets() argument
68 io::JoinPath(export_dir, kSavedModelAssetsDirectory); in ValidateAssets()
82 void CheckSavedModelBundle(const string& export_dir, in CheckSavedModelBundle() argument
84 ValidateAssets(export_dir, bundle); in CheckSavedModelBundle()
119 const string export_dir = in TEST_F() local
122 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
124 CheckSavedModelBundle(export_dir, bundle); in TEST_F()
133 const string export_dir = in TEST_F() local
135 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
137 CheckSavedModelBundle(export_dir, bundle); in TEST_F()
[all …]
Dsaved_model_bundle_lite_test.cc54 void ValidateAssets(const string& export_dir, in ValidateAssets() argument
57 io::JoinPath(export_dir, kSavedModelAssetsDirectory); in ValidateAssets()
71 void CheckSavedModelBundle(const string& export_dir, in CheckSavedModelBundle() argument
73 ValidateAssets(export_dir, bundle); in CheckSavedModelBundle()
117 const string export_dir = in TEST_F() local
120 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
122 CheckSavedModelBundle(export_dir, bundle); in TEST_F()
131 const string export_dir = in TEST_F() local
133 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
144 const string export_dir = in TEST_F() local
[all …]
Dreader_test.cc60 const string export_dir = GetDataDependencyFilepath(TestDataSharded()); in TEST_F() local
61 TF_ASSERT_OK(ReadMetaGraphDefFromSavedModel(export_dir, {kSavedModelTagServe}, in TEST_F()
69 const string export_dir = GetDataDependencyFilepath(TestDataSharded()); in TEST_F() local
70 Status st = ReadMetaGraphDefFromSavedModel(export_dir, {"missing-tag"}, in TEST_F()
82 const string export_dir = GetDataDependencyFilepath(TestDataSharded()); in TEST_F() local
84 export_dir, {kSavedModelTagServe, "missing-tag"}, &meta_graph_def); in TEST_F()
95 const string export_dir = GetDataDependencyFilepath(TestDataPbTxt()); in TEST_F() local
96 TF_ASSERT_OK(ReadMetaGraphDefFromSavedModel(export_dir, {kSavedModelTagServe}, in TEST_F()
104 const string export_dir = GetDataDependencyFilepath("missing-path"); in TEST_F() local
105 Status st = ReadMetaGraphDefFromSavedModel(export_dir, {kSavedModelTagServe}, in TEST_F()
[all …]
Dloader.cc125 void AddAssetsTensorsToInputs(const StringPiece export_dir, in AddAssetsTensorsToInputs() argument
133 export_dir, kSavedModelAssetsDirectory, asset_file_def.filename())); in AddAssetsTensorsToInputs()
189 Status RunInitOp(const RunOptions& run_options, const string& export_dir, in RunInitOp() argument
195 << export_dir; in RunInitOp()
197 AddAssetsTensorsToInputs(export_dir, asset_file_defs, &inputs); in RunInitOp()
205 Status RunRestore(const RunOptions& run_options, const string& export_dir, in RunRestore() argument
213 io::JoinPath(export_dir, kSavedModelVariablesDirectory); in RunRestore()
235 AddAssetsTensorsToInputs(export_dir, asset_file_defs, &inputs); in RunRestore()
258 const string& export_dir, in LoadSavedModelInternal() argument
261 TF_RETURN_IF_ERROR(ReadMetaGraphDefFromSavedModel(export_dir, tags, in LoadSavedModelInternal()
[all …]
Dreader.cc41 Status ReadSavedModel(absl::string_view export_dir, in ReadSavedModel() argument
43 LOG(INFO) << "Reading SavedModel from: " << export_dir; in ReadSavedModel()
46 io::JoinPath(export_dir, kSavedModelFilenamePb); in ReadSavedModel()
58 io::JoinPath(export_dir, kSavedModelFilenamePbTxt); in ReadSavedModel()
72 export_dir)); in ReadSavedModel()
135 Status ReadMetaGraphDefFromSavedModel(const string& export_dir, in ReadMetaGraphDefFromSavedModel() argument
139 TF_RETURN_IF_ERROR(ReadSavedModel(export_dir, &saved_model_proto)); in ReadMetaGraphDefFromSavedModel()
146 const string& export_dir, in ReadSavedModelDebugInfoIfPresent() argument
149 << export_dir; in ReadSavedModelDebugInfoIfPresent()
152 io::JoinPath(export_dir, "debug", "saved_model_debug_info.pb"); in ReadSavedModelDebugInfoIfPresent()
Dbundle_v2.cc38 Status ReadSavedModelProto(const string& export_dir, in ReadSavedModelProto() argument
40 LOG(INFO) << "Reading SavedModel from: " << export_dir; in ReadSavedModelProto()
43 io::JoinPath(export_dir, kSavedModelFilenamePb); in ReadSavedModelProto()
55 io::JoinPath(export_dir, kSavedModelFilenamePbTxt); in ReadSavedModelProto()
69 export_dir); in ReadSavedModelProto()
98 Status SavedModelV2Bundle::Load(const std::string& export_dir, in Load() argument
102 TF_RETURN_IF_ERROR(ReadSavedModelProto(export_dir, &saved_model_proto)); in Load()
118 ReadSavedModelDebugInfoIfPresent(export_dir, &bundle->debug_info_)); in Load()
121 io::JoinPath(export_dir, kSavedModelVariablesDirectory); in Load()
Dbundle_v2_test.cc74 const string export_dir = io::JoinPath( in TEST_F() local
78 TF_ASSERT_OK(SavedModelV2Bundle::Load(export_dir, &bundle)); in TEST_F()
87 const string export_dir = in TEST_F() local
91 TF_ASSERT_OK(SavedModelV2Bundle::Load(export_dir, &bundle)); in TEST_F()
104 const string export_dir = io::JoinPath( in TEST_F() local
108 TF_ASSERT_OK(SavedModelV2Bundle::Load(export_dir, &bundle)); in TEST_F()
Dloader.h105 const MetaGraphDef& meta_graph, const string& export_dir,
122 const RunOptions& run_options, const string& export_dir,
134 const RunOptions& run_options, const string& export_dir,
143 bool MaybeSavedModelDirectory(const std::string& export_dir);
Dreader.h33 Status ReadMetaGraphDefFromSavedModel(const string& export_dir,
39 const string& export_dir,
/external/tensorflow/tensorflow/python/saved_model/
Dsaved_model_test.py109 export_dir, argument
116 compat.as_bytes(export_dir),
191 export_dir = self._get_export_dir("test_bad_saved_model_file_format")
195 IOError, "SavedModel file does not exist at: %s" % export_dir):
196 loader.load(sess, ["foo"], export_dir)
198 os.makedirs(export_dir)
200 path_to_pb = os.path.join(export_dir, constants.SAVED_MODEL_FILENAME_PB)
206 loader.load(sess, ["foo"], export_dir)
209 file_io.delete_recursively(export_dir)
211 os.makedirs(export_dir)
[all …]
Dloader_impl.py47 def parse_saved_model_with_debug_info(export_dir): argument
60 saved_model = _parse_saved_model(export_dir)
63 saved_model_utils.get_debug_dir(export_dir),
77 def parse_saved_model(export_dir): argument
92 compat.as_bytes(compat.path_to_str(export_dir)),
96 compat.as_bytes(compat.path_to_str(export_dir)),
120 (export_dir, os.path.sep, constants.SAVED_MODEL_FILENAME_PBTXT,
129 def get_asset_tensors(export_dir, meta_graph_def_to_load, import_scope=None): argument
159 compat.as_bytes(export_dir), compat.as_bytes(constants.ASSETS_DIRECTORY))
237 def maybe_saved_model_directory(export_dir): argument
[all …]
Dutils_impl.py223 def get_or_create_variables_dir(export_dir): argument
225 variables_dir = get_variables_dir(export_dir)
230 def get_variables_dir(export_dir): argument
233 compat.as_text(export_dir),
237 def get_variables_path(export_dir): argument
240 compat.as_text(get_variables_dir(export_dir)),
244 def get_or_create_assets_dir(export_dir): argument
246 assets_destination_dir = get_assets_dir(export_dir)
253 def get_assets_dir(export_dir): argument
256 compat.as_text(export_dir),
[all …]
Dbuilder_impl.py97 def __init__(self, export_dir): argument
102 self._export_dir = export_dir
103 if file_io.file_exists(export_dir):
104 if file_io.list_directory(export_dir):
108 "specified directory: %s" % export_dir)
442 def __init__(self, export_dir): argument
443 super(SavedModelBuilder, self).__init__(export_dir=export_dir)
Dmethod_name_updater.py62 def __init__(self, export_dir): argument
72 self._export_dir = export_dir
73 self._saved_model = loader.parse_saved_model(export_dir)
Dpywrap_saved_model.cc32 m.def("Save", [](const char* export_dir) { in PYBIND11_MODULE() argument
33 MaybeRaiseFromStatus(libexport::Save(export_dir)); in PYBIND11_MODULE()
Dsimple_save.py35 def simple_save(session, export_dir, inputs, outputs, legacy_init_op=None): argument
83 b = builder.SavedModelBuilder(export_dir)
Dload.py128 def __init__(self, object_graph_proto, saved_model_proto, export_dir, argument
135 self._export_dir = export_dir
675 def load_partial(export_dir, filters, tags=None, options=None): argument
769 return load_internal(export_dir, tags, options, filters=filters)
773 def load(export_dir, tags=None, options=None): argument
863 result = load_internal(export_dir, tags, options)["root"]
867 def load_internal(export_dir, tags=None, options=None, loader_cls=Loader, argument
876 loader_impl.parse_saved_model_with_debug_info(export_dir))
901 loader = loader_cls(object_graph_proto, saved_model_proto, export_dir,
921 root = load_v1_in_v2.load(export_dir, tags)
/external/tensorflow/tensorflow/python/distribute/integration_test/
Dsaved_model_test.py91 export_dir = os.path.join(self._get_tempdir_path_test(),
98 tf.saved_model.save(m, export_dir)
146 export_dir = self.get_temp_dir()
154 tf.saved_model.save(m, export_dir)
156 loaded = tf.saved_model.load(export_dir)
177 export_dir = self.get_temp_dir()
181 tf.saved_model.save(m, export_dir)
183 loaded = tf.saved_model.load(export_dir)
204 export_dir = self.get_temp_dir()
207 tf.saved_model.save(m, export_dir)
[all …]
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_experimental.py436 def _get_or_create_variables_dir(export_dir): argument
438 variables_dir = _get_variables_dir(export_dir)
443 def _get_variables_dir(export_dir): argument
446 compat.as_text(export_dir),
450 def _get_variables_path(export_dir): argument
453 compat.as_text(_get_variables_dir(export_dir)),
457 def _get_or_create_assets_dir(export_dir): argument
459 assets_destination_dir = _get_assets_dir(export_dir)
466 def _get_assets_dir(export_dir): argument
469 compat.as_text(export_dir),
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.saved_model.pbtxt169 argspec: "args=[\'export_dir\'], varargs=None, keywords=None, defaults=None"
181 …argspec: "args=[\'sess\', \'tags\', \'export_dir\', \'import_scope\'], varargs=None, keywords=save…
185 …argspec: "args=[\'export_dir\', \'tags\', \'options\'], varargs=None, keywords=None, defaults=[\'N…
193 argspec: "args=[\'export_dir\'], varargs=None, keywords=None, defaults=None"
205 …argspec: "args=[\'obj\', \'export_dir\', \'signatures\', \'options\'], varargs=None, keywords=None…
209 …argspec: "args=[\'session\', \'export_dir\', \'inputs\', \'outputs\', \'legacy_init_op\'], varargs…
Dtensorflow.saved_model.loader.pbtxt5 …argspec: "args=[\'sess\', \'tags\', \'export_dir\', \'import_scope\'], varargs=None, keywords=save…
9 argspec: "args=[\'export_dir\'], varargs=None, keywords=None, defaults=None"
/external/tensorflow/tensorflow/cc/experimental/libexport/
Dsave.cc22 Status Save(const std::string& export_dir) { in Save() argument
23 TF_RETURN_IF_ERROR(Env::Default()->RecursivelyCreateDir(export_dir)); in Save()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.saved_model.pbtxt117 argspec: "args=[\'export_dir\'], varargs=None, keywords=None, defaults=None"
121 …argspec: "args=[\'export_dir\', \'tags\', \'options\'], varargs=None, keywords=None, defaults=[\'N…
125 …argspec: "args=[\'obj\', \'export_dir\', \'signatures\', \'options\'], varargs=None, keywords=None…
Dtensorflow.__internal__.saved_model.pbtxt17 …argspec: "args=[\'export_dir\', \'filters\', \'tags\', \'options\'], varargs=None, keywords=None, …
21 argspec: "args=[\'export_dir\'], varargs=None, keywords=None, defaults=None"
/external/tensorflow/tensorflow/java/src/main/native/
Dsaved_model_bundle_jni.cc24 JNIEnv* env, jclass clazz, jstring export_dir, jobjectArray tags, in Java_org_tensorflow_SavedModelBundle_load() argument
54 const char* cexport_dir = env->GetStringUTFChars(export_dir, nullptr); in Java_org_tensorflow_SavedModelBundle_load()
76 env->ReleaseStringUTFChars(export_dir, cexport_dir); in Java_org_tensorflow_SavedModelBundle_load()

123