Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/saved_model/
Dsaved_model_test.py85 export_dir, argument
92 compat.as_bytes(export_dir),
170 export_dir = self._get_export_dir("test_bad_saved_model_file_format")
175 export_dir):
176 loader.load(sess, ["foo"], export_dir)
178 os.makedirs(export_dir)
180 path_to_pb = os.path.join(export_dir, constants.SAVED_MODEL_FILENAME_PB)
186 loader.load(sess, ["foo"], export_dir)
189 file_io.delete_recursively(export_dir)
191 os.makedirs(export_dir)
[all …]
Dloader_impl.py42 def parse_saved_model(export_dir): argument
56 compat.as_bytes(export_dir),
60 compat.as_bytes(export_dir),
81 (export_dir,
91 def get_asset_tensors(export_dir, meta_graph_def_to_load, import_scope=None): argument
121 compat.as_bytes(export_dir), compat.as_bytes(constants.ASSETS_DIRECTORY))
199 def maybe_saved_model_directory(export_dir): argument
214 txt_path = os.path.join(export_dir, constants.SAVED_MODEL_FILENAME_PBTXT)
215 pb_path = os.path.join(export_dir, constants.SAVED_MODEL_FILENAME_PB)
220 def contains_saved_model(export_dir): argument
[all …]
Dutils_impl.py179 def get_or_create_variables_dir(export_dir): argument
181 variables_dir = get_variables_dir(export_dir)
187 def get_variables_dir(export_dir): argument
190 compat.as_text(export_dir),
194 def get_variables_path(export_dir): argument
197 compat.as_text(get_variables_dir(export_dir)),
201 def get_or_create_assets_dir(export_dir): argument
203 assets_destination_dir = get_assets_dir(export_dir)
211 def get_assets_dir(export_dir): argument
214 compat.as_text(export_dir),
Dload.py47 def __init__(self, object_graph_proto, saved_model_proto, export_dir): argument
53 self._export_dir = export_dir
301 def load(export_dir, tags=None): argument
345 saved_model_proto = loader_impl.parse_saved_model(export_dir)
355 .format(export_dir, meta_graph_def.meta_info_def.tags, tags))
360 export_dir)
364 root = load_v1_in_v2.load(export_dir, tags)
Dbuilder_impl.py91 def __init__(self, export_dir): argument
96 self._export_dir = export_dir
97 if file_io.file_exists(export_dir):
98 if file_io.list_directory(export_dir):
102 "specified directory: %s" % export_dir)
426 def __init__(self, export_dir): argument
427 super(SavedModelBuilder, self).__init__(export_dir=export_dir)
/external/tensorflow/tensorflow/cc/saved_model/
Dloader_test.cc53 void ValidateAssets(const string& export_dir, in ValidateAssets() argument
56 io::JoinPath(export_dir, kSavedModelAssetsDirectory); in ValidateAssets()
70 void CheckSavedModelBundle(const string& export_dir, in CheckSavedModelBundle() argument
72 ValidateAssets(export_dir, bundle); in CheckSavedModelBundle()
108 const string export_dir = in TEST_F() local
111 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
113 CheckSavedModelBundle(export_dir, bundle); in TEST_F()
122 const string export_dir = in TEST_F() local
124 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
126 CheckSavedModelBundle(export_dir, bundle); in TEST_F()
[all …]
Dloader.cc82 void AddAssetsTensorsToInputs(const StringPiece export_dir, in AddAssetsTensorsToInputs() argument
90 export_dir, kSavedModelAssetsDirectory, asset_file_def.filename())); in AddAssetsTensorsToInputs()
146 Status RunInitOp(const RunOptions& run_options, const string& export_dir, in RunInitOp() argument
153 AddAssetsTensorsToInputs(export_dir, asset_file_defs, &inputs); in RunInitOp()
164 Status GetInitOp(const string& export_dir, const MetaGraphDef& meta_graph_def, in GetInitOp() argument
189 strings::StrCat("Expected exactly one main op in : ", export_dir)); in GetInitOp()
196 Status RunRestore(const RunOptions& run_options, const string& export_dir, in RunRestore() argument
204 io::JoinPath(export_dir, kSavedModelVariablesDirectory); in RunRestore()
226 AddAssetsTensorsToInputs(export_dir, asset_file_defs, &inputs); in RunRestore()
261 const string& export_dir, in LoadSavedModelInternal() argument
[all …]
Dreader_test.cc52 const string export_dir = in TEST_F() local
54 TF_ASSERT_OK(ReadMetaGraphDefFromSavedModel(export_dir, {kSavedModelTagServe}, in TEST_F()
62 const string export_dir = in TEST_F() local
64 Status st = ReadMetaGraphDefFromSavedModel(export_dir, {"missing-tag"}, in TEST_F()
76 const string export_dir = in TEST_F() local
79 export_dir, {kSavedModelTagServe, "missing-tag"}, &meta_graph_def); in TEST_F()
90 const string export_dir = in TEST_F() local
92 TF_ASSERT_OK(ReadMetaGraphDefFromSavedModel(export_dir, {kSavedModelTagServe}, in TEST_F()
100 const string export_dir = in TEST_F() local
102 Status st = ReadMetaGraphDefFromSavedModel(export_dir, {kSavedModelTagServe}, in TEST_F()
Dreader.cc30 Status ReadSavedModel(const string& export_dir, SavedModel* saved_model_proto) { in ReadSavedModel() argument
31 LOG(INFO) << "Reading SavedModel from: " << export_dir; in ReadSavedModel()
34 io::JoinPath(export_dir, kSavedModelFilenamePb); in ReadSavedModel()
40 io::JoinPath(export_dir, kSavedModelFilenamePbTxt); in ReadSavedModel()
48 export_dir); in ReadSavedModel()
79 Status ReadMetaGraphDefFromSavedModel(const string& export_dir, in ReadMetaGraphDefFromSavedModel() argument
83 TF_RETURN_IF_ERROR(ReadSavedModel(export_dir, &saved_model_proto)); in ReadMetaGraphDefFromSavedModel()
/external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
Dexport_test.py68 def _assert_export(self, export_monitor, export_dir, expected_signature): argument
69 self.assertTrue(gfile.Exists(export_dir))
72 saver.checkpoint_exists(os.path.join(export_dir, '00000001', 'export')),
74 export_dir, '00000001', 'export'))
76 saver.checkpoint_exists(os.path.join(export_dir, '00000010', 'export')),
78 export_dir, '00000010', 'export'))
80 six.b(os.path.join(export_dir, '00000010')),
84 os.path.join(export_dir, '00000010', 'export.meta'))
93 export_dir = os.path.join(tempfile.mkdtemp(), 'export')
95 every_n_steps=1, export_dir=export_dir, exports_to_keep=2)
[all …]
Dexport.py69 def _export_graph(graph, saver, checkpoint_path, export_dir, argument
87 return export.export(export_dir, training_util.get_global_step(),
270 export_dir, argument
277 export_dir=export_dir,
286 export_dir, argument
357 export_dir,
/external/tensorflow/tensorflow/contrib/session_bundle/
Dsession_bundle.cc65 Status GetMetaGraphDefFromExport(const StringPiece export_dir, in GetMetaGraphDefFromExport() argument
68 io::JoinPath(export_dir, kMetaGraphDefFilename); in GetMetaGraphDefFromExport()
80 void AddAssetsTensorsToInputs(const StringPiece export_dir, in AddAssetsTensorsToInputs() argument
88 io::JoinPath(export_dir, kAssetsDirectory, asset.filename())); in AddAssetsTensorsToInputs()
111 string GetVariablesFilename(const StringPiece export_dir) { in GetVariablesFilename() argument
116 ->FileExists(io::JoinPath(export_dir, kVariablesFilename)) in GetVariablesFilename()
123 ->FileExists(io::JoinPath(export_dir, kVariablesIndexFilename)) in GetVariablesFilename()
125 return io::JoinPath(export_dir, kVariablesFilename); in GetVariablesFilename()
127 return io::JoinPath(export_dir, kVariablesFilenamePattern); in GetVariablesFilename()
131 Status RunRestoreOp(const RunOptions& run_options, const StringPiece export_dir, in RunRestoreOp() argument
[all …]
Dsession_bundle.py39 def maybe_session_bundle_dir(export_dir): argument
49 meta_graph_filename = os.path.join(export_dir,
56 def load_session_bundle_from_path(export_dir, argument
82 meta_graph_filename = os.path.join(export_dir,
96 variables_index_filename = os.path.join(export_dir,
104 export_dir, constants.VARIABLES_FILENAME_PATTERN_V2)
109 variables_filename = os.path.join(export_dir, constants.VARIABLES_FILENAME)
113 variables_filename = os.path.join(export_dir,
128 assets_dir = os.path.join(export_dir, constants.ASSETS_DIRECTORY)
149 saver.restore(sess, os.path.join(export_dir, restore_files))
Dbundle_shim.py184 def _load_saved_model_from_session_bundle_path(export_dir, target, config): argument
207 meta_graph_filename = os.path.join(export_dir,
234 export_dir, target, config, meta_graph_def=metagraph_def)
238 def load_session_bundle_or_saved_model_bundle_from_path(export_dir, argument
266 if loader.maybe_saved_model_directory(export_dir):
268 metagraph_def = loader.load(sess, tags, export_dir)
269 elif session_bundle.maybe_session_bundle_dir(export_dir):
270 sess, metagraph_def = _load_saved_model_from_session_bundle_path(export_dir,
275 "specified export location: %s" % export_dir)
Dbundle_shim.cc356 const string& export_dir, in LoadSessionBundleOrSavedModelBundle() argument
362 if (MaybeSavedModelDirectory(export_dir)) { in LoadSessionBundleOrSavedModelBundle()
365 << export_dir; in LoadSessionBundleOrSavedModelBundle()
367 return LoadSavedModel(session_options, run_options, export_dir, in LoadSessionBundleOrSavedModelBundle()
369 } else if (IsPossibleExportDirectory(export_dir)) { in LoadSessionBundleOrSavedModelBundle()
374 << export_dir; in LoadSessionBundleOrSavedModelBundle()
379 session_options, run_options, export_dir, saved_model_bundle); in LoadSessionBundleOrSavedModelBundle()
389 export_dir)); in LoadSessionBundleOrSavedModelBundle()
Dsession_bundle.h63 const StringPiece export_dir,
73 const StringPiece export_dir, SessionBundle* bundle);
80 bool IsPossibleExportDirectory(const StringPiece export_dir);
Dexporter.py268 export_dir = os.path.join(
275 if gfile.Exists(export_dir):
277 "not allowed. Duplicate export dir: %s" % export_dir)
281 tmp_export_dir = compat.as_text(export_dir) + "-tmp"
300 gfile.Rename(tmp_export_dir, export_dir)
319 return export_dir
/external/tensorflow/tensorflow/examples/saved_model/integration_tests/
Dsaved_model_part1_test.py44 export_dir = self.get_temp_dir()
47 self.assertCommandSucceeded(export_binary, export_dir=export_dir)
50 self.assertCommandSucceeded(use_binary, model_dir=export_dir)
54 export_dir = self.get_temp_dir()
57 self.assertCommandSucceeded(export_binary, export_dir=export_dir)
60 self.assertCommandSucceeded(use_binary, model_dir=export_dir)
64 export_dir = self.get_temp_dir()
67 self.assertCommandSucceeded(export_binary, export_dir=export_dir)
71 self.assertCommandSucceeded(use_binary, model_dir=export_dir)
Dsaved_model_part2_test.py44 export_dir = self.get_temp_dir()
47 export_binary, export_dir=export_dir, fast_test_mode="true")
51 use_binary, export_dir=export_dir, fast_test_mode="true")
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dsvm.py197 def export(self, export_dir, signature_fn=None, argument
202 export_dir=export_dir,
211 export_dir, argument
220 return super(SVM, self).export(export_dir=export_dir,
Destimator_test.py974 export_dir = est.export_savedmodel(
978 self.assertTrue(gfile.Exists(export_dir))
982 compat.as_bytes(export_dir),
987 compat.as_bytes(export_dir), compat.as_bytes('variables'))))
991 compat.as_bytes(export_dir),
996 compat.as_bytes(export_dir),
1002 compat.as_bytes(export_dir), compat.as_bytes('assets'))))
1006 compat.as_bytes(export_dir),
1013 compat.as_bytes(export_dir),
1017 compat.as_bytes(export_dir),
[all …]
/external/tensorflow/tensorflow/contrib/predictor/
Dsaved_model_predictor.py61 def _get_signature_def(signature_def_key, export_dir, tags): argument
67 metagraph_def = get_meta_graph_def(export_dir, tags)
113 export_dir, argument
153 loader.load(self._session, tags.split(','), export_dir)
157 signature_def = _get_signature_def(signature_def_key, export_dir, tags)
Dsaved_model_predictor_test.py54 export_dir=self._export_dir)
67 export_dir=self._export_dir,
110 export_dir=self._export_dir,
136 export_dir=self._export_dir,
157 export_dir=self._export_dir,
164 export_dir=self._export_dir,
/external/jsoncpp/
Dmakerelease.py110 def svn_export( tag_url, export_dir ): argument
115 rmdir_if_exist( export_dir )
116 svn_command( 'export', tag_url, export_dir )
321 export_dir = 'dist/export'
322 svn_export( tag_url, export_dir )
323 fix_sources_eol( export_dir )
328 tarball.make_tarball( source_tarball_path, [export_dir], export_dir, prefix_dir=source_dir )
333 … amalgamate.amalgamate_source( export_dir, '%s/jsoncpp.cpp' % amalgamation_dir, 'json/json.h' )
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.saved_model.pbtxt153 argspec: "args=[\'export_dir\'], varargs=None, keywords=None, defaults=None"
165 …argspec: "args=[\'sess\', \'tags\', \'export_dir\', \'import_scope\'], varargs=None, keywords=save…
169 argspec: "args=[\'export_dir\', \'tags\'], varargs=None, keywords=None, defaults=[\'None\'], "
177 argspec: "args=[\'export_dir\'], varargs=None, keywords=None, defaults=None"
189 …argspec: "args=[\'obj\', \'export_dir\', \'signatures\'], varargs=None, keywords=None, defaults=[\…
193 …argspec: "args=[\'session\', \'export_dir\', \'inputs\', \'outputs\', \'legacy_init_op\'], varargs…

123