Home
last modified time | relevance | path

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

123

/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dsaved_model_test.py104 export_dir, argument
111 compat.as_bytes(export_dir),
186 export_dir = self._get_export_dir("test_bad_saved_model_file_format")
190 IOError, "SavedModel file does not exist at: %s" % export_dir):
191 loader.load(sess, ["foo"], export_dir)
193 os.makedirs(export_dir)
195 path_to_pb = os.path.join(export_dir, constants.SAVED_MODEL_FILENAME_PB)
201 loader.load(sess, ["foo"], export_dir)
204 file_io.delete_recursively(export_dir)
206 os.makedirs(export_dir)
[all …]
H A Dloader_impl.py44 def parse_saved_model_with_debug_info(export_dir): argument
57 saved_model = parse_saved_model(export_dir)
60 saved_model_utils.get_debug_dir(export_dir),
74 def parse_saved_model(export_dir): argument
89 compat.as_bytes(compat.path_to_str(export_dir)),
93 compat.as_bytes(compat.path_to_str(export_dir)),
121 def get_asset_tensors(export_dir, meta_graph_def_to_load, import_scope=None): argument
151 compat.as_bytes(export_dir), compat.as_bytes(constants.ASSETS_DIRECTORY))
229 def maybe_saved_model_directory(export_dir): argument
244 txt_path = file_io.join(export_dir, constants.SAVED_MODEL_FILENAME_PBTXT)
[all …]
H A Dutils_impl.py216 def get_or_create_variables_dir(export_dir): argument
218 variables_dir = get_variables_dir(export_dir)
223 def get_variables_dir(export_dir): argument
226 compat.as_text(export_dir), compat.as_text(constants.VARIABLES_DIRECTORY))
229 def get_variables_path(export_dir): argument
232 compat.as_text(get_variables_dir(export_dir)),
236 def get_or_create_assets_dir(export_dir): argument
238 assets_destination_dir = get_assets_dir(export_dir)
245 def get_assets_dir(export_dir): argument
248 compat.as_text(export_dir), compat.as_text(constants.ASSETS_DIRECTORY))
[all …]
H A Dsave.py1018 def _export_debug_info(exported_graph, export_dir): argument
1042 utils_impl.get_or_create_debug_dir(export_dir),
1050 def save(obj, export_dir, signatures=None, options=None): argument
1228 if isinstance(export_dir, os.PathLike):
1229 export_dir = os.fspath(export_dir)
1232 save_and_return_nodes(obj, export_dir, signatures, options)
1238 export_dir, argument
1275 utils_impl.get_or_create_variables_dir(export_dir)
1279 utils_impl.get_variables_path(export_dir), options=ckpt_options)
1281 export_dir)
[all …]
H A Dload.py145 def __init__(self, object_graph_proto, saved_model_proto, export_dir, argument
152 self._export_dir = export_dir
605 export_dir=self._export_dir,
723 def load(export_dir, tags=None, options=None): argument
813 if isinstance(export_dir, os.PathLike):
814 export_dir = os.fspath(export_dir)
815 result = load_partial(export_dir, None, tags, options)["root"]
820 def load_partial(export_dir, filters, tags=None, options=None): argument
920 loader_impl.parse_saved_model_with_debug_info(export_dir))
945 loader = Loader(object_graph_proto, saved_model_proto, export_dir,
[all …]
H A Dbuilder_impl.py93 def __init__(self, export_dir): argument
98 self._export_dir = export_dir
99 if file_io.file_exists(export_dir):
100 if file_io.list_directory(export_dir):
438 def __init__(self, export_dir): argument
439 super(SavedModelBuilder, self).__init__(export_dir=export_dir)
H A Dpywrap_saved_model_fingerprinting_test.py30 export_dir = test.test_src_dir_path(
32 with file_io.FileIO(os.path.join(export_dir, "saved_model.pb"), "rb") as f:
37 fingerprinting.CreateFingerprintDef(file_content, export_dir))
H A Dmethod_name_updater.py56 def __init__(self, export_dir): argument
66 self._export_dir = export_dir
67 self._saved_model = loader.parse_saved_model(export_dir)
H A Dsimple_save_test.py54 export_dir = os.path.join(test.get_temp_dir(),
69 simple_save.simple_save(sess, export_dir, inputs, outputs)
74 graph = loader.load(sess, [tag_constants.SERVING], export_dir)
/aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/
H A 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 …]
H A 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 …]
H A Dfingerprinting_test.cc65 const std::string export_dir = in TEST() local
69 ReadSavedModel(export_dir)); in TEST()
71 CreateFingerprintDef(saved_model_pb.meta_graphs(0), export_dir); in TEST()
85 const std::string export_dir = io::JoinPath( in TEST() local
88 ReadSavedModel(export_dir)); in TEST()
110 const std::string export_dir = io::JoinPath( in TEST() local
114 ReadSavedModel(export_dir)); in TEST()
116 CreateFingerprintDef(saved_model_pb.meta_graphs(0), export_dir); in TEST()
134 const std::string export_dir = io::JoinPath( in TEST() local
137 ReadSavedModel(export_dir)); in TEST()
[all …]
H A 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 …]
H A Dloader.cc144 void AddAssetsTensorsToInputs(const StringPiece export_dir, in AddAssetsTensorsToInputs() argument
152 export_dir, kSavedModelAssetsDirectory, asset_file_def.filename())); in AddAssetsTensorsToInputs()
208 Status RunInitOp(const RunOptions& run_options, const string& export_dir, in RunInitOp() argument
214 << export_dir; in RunInitOp()
216 AddAssetsTensorsToInputs(export_dir, asset_file_defs, &inputs); in RunInitOp()
224 Status RunRestore(const RunOptions& run_options, const string& export_dir, in RunRestore() argument
232 io::JoinPath(export_dir, kSavedModelVariablesDirectory); in RunRestore()
257 AddAssetsTensorsToInputs(export_dir, asset_file_defs, &inputs); in RunRestore()
280 const string& export_dir, in LoadSavedModelInternal() argument
283 TF_RETURN_IF_ERROR(ReadMetaGraphDefFromSavedModel(export_dir, tags, in LoadSavedModelInternal()
[all …]
H A Dreader.cc43 Status ReadSavedModel(absl::string_view export_dir, in ReadSavedModel() argument
45 LOG(INFO) << "Reading SavedModel from: " << export_dir; in ReadSavedModel()
48 io::JoinPath(export_dir, kSavedModelFilenamePb); in ReadSavedModel()
63 io::JoinPath(export_dir, kSavedModelFilenamePbTxt); in ReadSavedModel()
80 export_dir, in ReadSavedModel()
117 Status ReadMetaGraphDefFromSavedModel(const string& export_dir, in ReadMetaGraphDefFromSavedModel() argument
121 TF_RETURN_IF_ERROR(ReadSavedModel(export_dir, &saved_model_proto)); in ReadMetaGraphDefFromSavedModel()
128 const string& export_dir, in ReadSavedModelDebugInfoIfPresent() argument
131 << export_dir; in ReadSavedModelDebugInfoIfPresent()
134 io::JoinPath(export_dir, "debug", "saved_model_debug_info.pb"); in ReadSavedModelDebugInfoIfPresent()
H A Dbundle_v2.cc39 Status ReadSavedModelProto(const string& export_dir, in ReadSavedModelProto() argument
41 LOG(INFO) << "Reading SavedModel from: " << export_dir; in ReadSavedModelProto()
44 io::JoinPath(export_dir, kSavedModelFilenamePb); in ReadSavedModelProto()
56 io::JoinPath(export_dir, kSavedModelFilenamePbTxt); in ReadSavedModelProto()
70 export_dir); in ReadSavedModelProto()
99 Status SavedModelV2Bundle::Load(const std::string& export_dir, in Load() argument
103 TF_RETURN_IF_ERROR(ReadSavedModelProto(export_dir, &saved_model_proto)); in Load()
124 ReadSavedModelDebugInfoIfPresent(export_dir, &bundle->debug_info_)); in Load()
127 io::JoinPath(export_dir, kSavedModelVariablesDirectory); in Load()
H A 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()
H A 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);
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/integration_test/
H A Dsaved_model_test.py88 export_dir = os.path.join(self._get_tempdir_path_test(),
95 tf.saved_model.save(m, export_dir)
144 export_dir = self.get_temp_dir()
152 tf.saved_model.save(m, export_dir)
154 loaded = tf.saved_model.load(export_dir)
175 export_dir = self.get_temp_dir()
179 tf.saved_model.save(m, export_dir)
181 loaded = tf.saved_model.load(export_dir)
202 export_dir = self.get_temp_dir()
205 tf.saved_model.save(m, export_dir)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/saving/
H A Dsaved_model_experimental.py435 def _get_or_create_variables_dir(export_dir): argument
437 variables_dir = _get_variables_dir(export_dir)
442 def _get_variables_dir(export_dir): argument
445 compat.as_text(export_dir),
449 def _get_variables_path(export_dir): argument
452 compat.as_text(_get_variables_dir(export_dir)),
456 def _get_or_create_assets_dir(export_dir): argument
458 assets_destination_dir = _get_assets_dir(export_dir)
465 def _get_assets_dir(export_dir): argument
468 compat.as_text(export_dir),
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A 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…
/aosp_15_r20/external/pciutils/lib/
H A Dphysmem-windows.c407 IMAGE_EXPORT_DIRECTORY *export_dir; in win32_get_proc_address_by_ordinal() local
444 export_dir = (IMAGE_EXPORT_DIRECTORY *)(baseaddr + export_dir_offset); in win32_get_proc_address_by_ordinal()
445 base_ordinal = export_dir->Base; in win32_get_proc_address_by_ordinal()
446 func_count = export_dir->NumberOfFunctions; in win32_get_proc_address_by_ordinal()
447 func_addrs = (DWORD *)(baseaddr + (DWORD)export_dir->AddressOfFunctions); in win32_get_proc_address_by_ordinal()
455 names_count = export_dir->NumberOfNames; in win32_get_proc_address_by_ordinal()
456 names_idxs = (USHORT *)(baseaddr + (DWORD)export_dir->AddressOfNameOrdinals); in win32_get_proc_address_by_ordinal()
465 …if ((BYTE *)func_ptr >= (BYTE *)export_dir && (BYTE *)func_ptr < (BYTE *)export_dir + export_dir_s… in win32_get_proc_address_by_ordinal()
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A 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…
/aosp_15_r20/external/tensorflow/tensorflow/java/src/main/native/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/model_utils/
H A Dexport_test.py241 export_dir = os.path.join("tmp", "export", "1576013284")
242 tmp_export_dir = export_utils.get_temp_export_dir(export_dir)
246 export_dir = os.path.join(b"tmp", b"export", b"1576013284")
247 tmp_export_dir = export_utils.get_temp_export_dir(export_dir)

123