Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/python/
Dconvert_saved_model_test.py122 saved_model_dir = os.path.join(self.get_temp_dir(), "simple_savedmodel")
128 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
129 return saved_model_dir
133 saved_model_dir = os.path.join(self.get_temp_dir(), "simple_savedmodel")
142 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
143 return saved_model_dir
161 saved_model_dir, argument
172 saved_model_dir=saved_model_dir,
182 saved_model_dir = self._createSimpleSavedModel(shape=[1, 16, 16, 3])
183 _, in_tensors, out_tensors = self._convertSavedModel(saved_model_dir)
[all …]
Dconvert_saved_model.py47 def get_meta_graph_def(saved_model_dir, tag_set): argument
61 return loader.load(sess, tag_set, saved_model_dir)
222 def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, argument
253 meta_graph = get_meta_graph_def(saved_model_dir, tag_set)
264 loader.load(sess, meta_graph.meta_info_def.tags, saved_model_dir)
Dlite_test.py883 saved_model_dir = os.path.join(self.get_temp_dir(), 'simple_savedmodel')
892 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
893 return saved_model_dir
897 saved_model_dir = self._createSavedModel(shape=[1, 16, 16, 3])
900 converter = lite.TFLiteConverter.from_saved_model(saved_model_dir)
928 saved_model_dir = self._createSavedModel(shape=[None, 16, 16, 3])
930 converter = lite.TFLiteConverter.from_saved_model(saved_model_dir)
959 saved_model_dir = self._createSavedModel(shape=[1, 16, 16, 3])
962 saved_model_dir, input_arrays=['inputB', 'inputA'])
991 saved_model_dir = self._createSavedModel(shape=[1, 16, 16, 3])
[all …]
Dlite.py567 saved_model_dir, argument
598 result = _freeze_saved_model(saved_model_dir, input_arrays, input_shapes,
846 saved_model_dir, argument
853 return TFLiteConverter.from_saved_model(saved_model_dir, input_arrays,
Dtflite_convert.py98 elif flags.saved_model_dir:
100 converter_kwargs["saved_model_dir"] = flags.saved_model_dir
/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
88 saved_model = read_saved_model(saved_model_dir)
95 def get_meta_graph_def(saved_model_dir, tag_set): argument
113 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.py50 def _show_tag_sets(saved_model_dir): argument
58 tag_sets = saved_model_utils.get_saved_model_tag_sets(saved_model_dir)
64 def _show_signature_def_map_keys(saved_model_dir, tag_set): argument
76 signature_def_map = get_signature_def_map(saved_model_dir, tag_set)
119 def _show_inputs_outputs(saved_model_dir, tag_set, signature_def_key, indent=0): argument
132 meta_graph_def = saved_model_utils.get_meta_graph_def(saved_model_dir,
183 def _show_all(saved_model_dir): argument
192 tag_sets = saved_model_utils.get_saved_model_tag_sets(saved_model_dir)
198 signature_def_map = get_signature_def_map(saved_model_dir, tag_set)
201 _show_inputs_outputs(saved_model_dir, tag_set, signature_def_key,
[all …]
Dfreeze_graph_test.py223 saved_model_dir = os.path.join(tmp_dir, "saved_model_dir")
225 self._writeDummySavedModel(saved_model_dir, feature_name)
228 input_saved_model_dir = saved_model_dir
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_test.py71 saved_model_dir = self._save_model_dir()
72 keras_saved_model.export_saved_model(model, saved_model_dir)
74 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)
134 saved_model_dir = self._save_model_dir()
[all …]
/external/tensorflow/tensorflow/contrib/saved_model/python/saved_model/
Dreader_test.py44 saved_model_dir = os.path.join(test.get_temp_dir(), "valid_saved_model")
45 builder = saved_model_builder.SavedModelBuilder(saved_model_dir)
51 actual_saved_model_pb = reader.read_saved_model(saved_model_dir)
59 saved_model_dir = os.path.join(test.get_temp_dir(), "invalid_saved_model")
61 IOError, "SavedModel file does not exist at: %s" % saved_model_dir):
62 reader.read_saved_model(saved_model_dir)
65 saved_model_dir = os.path.join(test.get_temp_dir(), "test_tags")
66 builder = saved_model_builder.SavedModelBuilder(saved_model_dir)
106 actual_tags = reader.get_saved_model_tag_sets(saved_model_dir)
Dreader.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
88 saved_model = read_saved_model(saved_model_dir)
/external/tensorflow/tensorflow/contrib/predictor/
DREADME.md34 saved_model_dir = estimator.export_savedmodel(my_export_dir, serving_input_fn)
60 export_dir=saved_model_dir,
72 export_dir=saved_model_dir,
Dsaved_model_predictor.py37 def get_meta_graph_def(saved_model_dir, tags): argument
53 saved_model = reader.read_saved_model(saved_model_dir)
/external/tensorflow/tensorflow/lite/testing/model_coverage/
Dmodel_coverage_lib_test.py113 saved_model_dir = os.path.join(self.get_temp_dir(), 'simple_savedmodel')
123 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
124 model_coverage.test_saved_model(saved_model_dir)
/external/tensorflow/tensorflow/lite/g3doc/convert/
Dquantization.md16 converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
31 converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
Dcmdline_reference.md13 `--graph_def_file`, `--saved_model_dir` or `--keras_model_file` is required.
19 * `--saved_model_dir`. Type: string. Specifies the full path to the directory
Dpython_api.md91 converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.lite.-toco-converter.pbtxt18 …argspec: "args=[\'cls\', \'saved_model_dir\', \'input_arrays\', \'input_shapes\', \'output_arrays\…
Dtensorflow.lite.-t-f-lite-converter.pbtxt23 …argspec: "args=[\'cls\', \'saved_model_dir\', \'input_arrays\', \'input_shapes\', \'output_arrays\…
/external/tensorflow/tensorflow/lite/tools/optimize/g3doc/
Dquantize_weights.md15 --saved_model_dir=/tmp/saved_model \
/external/tensorflow/tensorflow/lite/tutorials/
Dpost_training_quant.ipynb227 "saved_model_dir = str(sorted(pathlib.Path(saved_models_root).glob(\"*\"))[-1])\n",
228 "saved_model_dir"
255 "converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)\n",
/external/tensorflow/tensorflow/lite/g3doc/performance/
Dpost_training_quantization.md10 converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dget_started.md246 conversion tool. Assuming that the saved model is stored in saved_model_dir, the
250 converter=tf.contrib.lite.TocoConverter.from_saved_model(saved_model_dir)
/external/tensorflow/tensorflow/c/
Dc_api_test.cc1354 const string saved_model_dir = tensorflow::io::JoinPath( in TEST() local
1363 opt, run_options, saved_model_dir.c_str(), tags, 1, graph, metagraph, s); in TEST()
1429 const string saved_model_dir = tensorflow::io::JoinPath( in TEST() local
1437 opt, nullptr, saved_model_dir.c_str(), tags, 1, graph, nullptr, s); in TEST()

12