Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/python/
Dconvert_saved_model_test.py44 saved_model_dir = os.path.join(self.get_temp_dir(), "simple_savedmodel")
50 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
51 return saved_model_dir
55 saved_model_dir = os.path.join(self.get_temp_dir(), "simple_savedmodel")
64 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
65 return saved_model_dir
83 saved_model_dir, argument
95 saved_model_dir=saved_model_dir,
105 saved_model_dir = self._createSimpleSavedModel(shape=[1, 16, 16, 3])
106 _, in_tensors, out_tensors = self._convertSavedModel(saved_model_dir)
[all …]
Dlite.py443 self.saved_model_dir = None
518 if self.saved_model_dir:
520 "saved_model_dir": self.saved_model_dir,
544 self.saved_model_dir = None
546 if self.saved_model_dir:
549 _parse_saved_model_with_debug_info(self.saved_model_dir))
553 self.saved_model_dir = None
557 self.saved_model_dir = None
564 self.saved_model_dir = None
698 saved_model_dir, argument
[all …]
Dtflite_convert_test.py230 saved_model_dir = self._getFilepath('model')
238 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
240 flags_str = '--saved_model_dir={}'.format(saved_model_dir)
271 saved_model_dir = self._getFilepath('model')
275 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
276 return (saved_model_dir, custom_opdefs_str)
279 saved_model_dir, _ = self._createSavedModelWithCustomOp()
283 '--experimental_new_converter'.format(saved_model_dir))
287 saved_model_dir, custom_opdefs_str = self._createSavedModelWithCustomOp()
292 '--experimental_new_converter'.format(saved_model_dir,
[all …]
Dconvert_saved_model.py46 def get_meta_graph_def(saved_model_dir, tag_set): argument
60 return loader.load(sess, tag_set, saved_model_dir)
155 def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, argument
187 meta_graph = get_meta_graph_def(saved_model_dir, tag_set)
198 loader.load(sess, meta_graph.meta_info_def.tags, saved_model_dir)
Dconvert.py385 saved_model_dir=None, argument
493 if saved_model_dir:
523 if saved_model_dir:
530 if saved_model_dir:
531 model.saved_model_dir = saved_model_dir
636 def convert_saved_model(saved_model_dir=None, argument
643 if saved_model_dir:
644 model_flags.saved_model_dir = saved_model_dir
Dlite_flex_test.py182 saved_model_dir = os.path.join(self.get_temp_dir(), 'model')
186 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
188 converter = lite.TFLiteConverterV2.from_saved_model(saved_model_dir)
197 saved_model_dir = os.path.join(self.get_temp_dir(), 'model2')
205 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
207 converter = lite.TFLiteConverterV2.from_saved_model(saved_model_dir)
Dlite_v2_test.py752 saved_model_dir = os.path.join(self.get_temp_dir(), 'simple_savedmodel')
764 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
765 return saved_model_dir
771 saved_model_dir = self._createV1SavedModel(shape=[1, 16, 16, 3])
774 converter = lite.TFLiteConverterV2.from_saved_model(saved_model_dir)
806 saved_model_dir = self._createV1SavedModel(shape=[1, 16, 16, 3])
810 saved_model_proto = parse_saved_model(saved_model_dir)
813 saved_model_pb_file_path = os.path.join(saved_model_dir, 'saved_model.pb')
818 converter = lite.TFLiteConverterV2.from_saved_model(saved_model_dir)
825 saved_model_dir = os.path.join(self.get_temp_dir(),
[all …]
/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dsaved_model_test.py153 saved_model_dir = self._save_model_dir()
154 tf_save.save(model, saved_model_dir)
155 loaded = keras_load.load(saved_model_dir)
196 saved_model_dir = self._save_model_dir()
198 tf_save.save(layer, saved_model_dir)
199 loaded = keras_load.load(saved_model_dir)
225 saved_model_dir = self._save_model_dir()
226 tf_save.save(model, saved_model_dir)
237 tf_save.save(model, saved_model_dir)
243 saved_model_dir = self._save_model_dir()
[all …]
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_utils.py31 def read_saved_model(saved_model_dir): argument
45 compat.as_bytes(saved_model_dir),
49 compat.as_bytes(saved_model_dir),
55 raise IOError("SavedModel file does not exist at: %s" % saved_model_dir)
75 (saved_model_dir, constants.SAVED_MODEL_FILENAME_PBTXT,
79 def get_saved_model_tag_sets(saved_model_dir): argument
89 saved_model = read_saved_model(saved_model_dir)
96 def get_meta_graph_def(saved_model_dir, tag_set): argument
115 saved_model = read_saved_model(saved_model_dir)
Dsaved_model_utils_test.py46 saved_model_dir = os.path.join(test.get_temp_dir(), "valid_saved_model")
47 builder = saved_model_builder.SavedModelBuilder(saved_model_dir)
53 actual_saved_model_pb = saved_model_utils.read_saved_model(saved_model_dir)
61 saved_model_dir = os.path.join(test.get_temp_dir(), "invalid_saved_model")
63 IOError, "SavedModel file does not exist at: %s" % saved_model_dir):
64 saved_model_utils.read_saved_model(saved_model_dir)
68 saved_model_dir = os.path.join(test.get_temp_dir(), "test_tags")
69 builder = saved_model_builder.SavedModelBuilder(saved_model_dir)
109 actual_tags = saved_model_utils.get_saved_model_tag_sets(saved_model_dir)
Dsaved_model_cli.py65 def _show_tag_sets(saved_model_dir): argument
73 tag_sets = saved_model_utils.get_saved_model_tag_sets(saved_model_dir)
79 def _show_signature_def_map_keys(saved_model_dir, tag_set): argument
91 signature_def_map = get_signature_def_map(saved_model_dir, tag_set)
134 def _show_inputs_outputs(saved_model_dir, tag_set, signature_def_key, indent=0): argument
147 meta_graph_def = saved_model_utils.get_meta_graph_def(saved_model_dir,
173 def _show_defined_functions(saved_model_dir): argument
179 meta_graphs = saved_model_utils.read_saved_model(saved_model_dir).meta_graphs
187 trackable_object = load.load(saved_model_dir)
287 def _show_all(saved_model_dir): argument
[all …]
/external/tensorflow/tensorflow/python/keras/saving/
Dsave_test.py364 saved_model_dir = self._save_model_dir()
402 model, saved_model_dir, save_format=save_format,
405 loaded_model = keras.models.load_model(saved_model_dir)
417 saved_model_dir = self._save_model_dir()
441 model.save(saved_model_dir, save_format=save_format)
443 new_model = keras.models.load_model(saved_model_dir)
452 saved_model_dir = self._save_model_dir()
464 keras.models.save_model(model, saved_model_dir, save_format=save_format)
466 new_model = keras.models.load_model(saved_model_dir)
473 saved_model_dir = self._save_model_dir()
[all …]
Dsaved_model_experimental_test.py72 saved_model_dir = self._save_model_dir()
73 keras_saved_model.export_saved_model(model, saved_model_dir)
75 loaded_model = keras_saved_model.load_from_saved_model(saved_model_dir)
89 saved_model_dir = self._save_model_dir()
90 keras_saved_model.export_saved_model(model, saved_model_dir)
91 loaded_model = keras_saved_model.load_from_saved_model(saved_model_dir)
113 saved_model_dir = self._save_model_dir()
114 keras_saved_model.export_saved_model(model, saved_model_dir)
115 loaded_model = keras_saved_model.load_from_saved_model(saved_model_dir)
133 saved_model_dir = self._save_model_dir()
[all …]
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py366 def _RunGraphV1(self, saved_model_dir, inputs_data, config, num_runs=2): argument
373 loader.load(sess, [tag_constants.SERVING], saved_model_dir)
392 def _RunGraphV2(self, saved_model_dir, inputs_data, graph_state, num_runs=2): argument
395 root = load.load(saved_model_dir)
428 saved_model_dir, argument
437 results = self._RunGraphV2(saved_model_dir, inputs_data, graph_state,
449 return self._RunGraphV1(saved_model_dir, inputs_data, config, num_runs)
451 def _CreateConverter(self, run_params, saved_model_dir, conversion_params): argument
455 input_saved_model_dir=saved_model_dir,
464 input_saved_model_dir=saved_model_dir,
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dtf_mlir_translate.cc149 absl::string_view saved_model_dir, in SavedModelObjectGraphToMlirImport() argument
154 std::string(saved_model_dir.data(), saved_model_dir.length()), &bundle); in SavedModelObjectGraphToMlirImport()
156 LOG(ERROR) << "Failed to load saved model '" << saved_model_dir in SavedModelObjectGraphToMlirImport()
169 absl::string_view saved_model_dir, in SavedModelSignatureDefsToMlirImport() argument
179 std::string(saved_model_dir), tags, &bundle); in SavedModelSignatureDefsToMlirImport()
181 LOG(ERROR) << "Failed to load saved model v1 '" << saved_model_dir in SavedModelSignatureDefsToMlirImport()
195 absl::string_view saved_model_dir, in SavedModelSignatureDefsToMlirImportLite() argument
200 auto status = ReadMetaGraphDefFromSavedModel(std::string(saved_model_dir), in SavedModelSignatureDefsToMlirImportLite()
203 LOG(ERROR) << "Failed to load saved model v1 '" << saved_model_dir in SavedModelSignatureDefsToMlirImportLite()
Dtf_mlir_translate.h97 absl::string_view saved_model_dir,
105 absl::string_view saved_model_dir,
115 absl::string_view saved_model_dir,
/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
Drun_models.py83 def __init__(self, saved_model_dir: str, saved_model_tags: Sequence[str],
94 saved_model_dir=saved_model_dir,
137 model_name = os.path.split(manager.model_config.saved_model_dir)[-1]
197 saved_model_dir=FLAGS.saved_model_dir,
Dmodel_handler.py115 saved_model_dir: str, saved_model_tags: str,
122 export_dir=saved_model_dir,
137 def load_graph_func(saved_model_dir: str, saved_model_tags: str,
141 export_dir=saved_model_dir, tags=saved_model_tags)
155 saved_model_dir: str,
161 cls).__new__(cls, saved_model_dir, saved_model_tags,
267 saved_model_dir=self.model_config.saved_model_dir,
337 saved_model_dir=self.model_config.saved_model_dir,
444 saved_model_dir=output_saved_model_dir)
454 input_saved_model_dir=self.model_config.saved_model_dir,
[all …]
/external/tensorflow/tensorflow/lite/testing/model_coverage/
Dmodel_coverage_lib_test.py144 saved_model_dir = os.path.join(self.get_temp_dir(), 'simple_savedmodel')
155 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
156 model_coverage.test_saved_model(saved_model_dir)
160 saved_model_dir = os.path.join(self.get_temp_dir(), 'simple_savedmodel')
184 saved_model.save(model, saved_model_dir)
187 saved_model_dir,
/external/tensorflow/tensorflow/python/keras/tests/
Dmodel_architectures_test.py89 saved_model_dir = self._save_model_dir()
99 keras.models.save_model(model, saved_model_dir, save_format=save_format)
102 saved_model_dir,
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Dasset.cc33 const std::string& saved_model_dir, in Create() argument
37 io::JoinPath(saved_model_dir, kSavedModelAssetsDirectory, asset_filename); in Create()
/external/tensorflow/tensorflow/cc/saved_model/experimental/tests/
Dsaved_model_api_test.cc39 std::string SavedModelPath(tensorflow::StringPiece saved_model_dir) { in SavedModelPath() argument
41 kTestData, saved_model_dir); in SavedModelPath()
/external/tensorflow/tensorflow/python/keras/utils/
Dkpl_test_utils.py175 saved_model_dir = tempfile.mkdtemp(dir=self.get_temp_dir())
177 model, saved_model_dir, signatures={"serving_default": serving_fn})
181 saved_model_dir).signatures["serving_default"]
/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
Drestore_ops_test.cc32 std::string CheckpointPrefix(StringPiece saved_model_dir) { in CheckpointPrefix() argument
34 saved_model_dir, kSavedModelVariablesDirectory, in CheckpointPrefix()
/external/tensorflow/tensorflow/c/experimental/saved_model/internal/
Dsaved_model_api.cc41 std::string saved_model_dir(dirname); in TF_LoadSavedModel() local
66 std::string saved_model_dir(dirname); in TF_LoadSavedModelWithTags() local

123