| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| D | BUILD | 10 # C API functions exposed in tf/c/experimental/saved_model/public/.
33 "//tensorflow/c/experimental/saved_model/public:concrete_function.h",
37 "//tensorflow/c/experimental/saved_model/public:__pkg__",
50 "//tensorflow/c/experimental/saved_model/core:concrete_function",
51 "//tensorflow/c/experimental/saved_model/core:function_metadata",
63 "//tensorflow/c/experimental/saved_model/public:concrete_function_list.h",
67 "//tensorflow/c/experimental/saved_model/public:__pkg__",
74 "//tensorflow/c/experimental/saved_model/core:concrete_function",
84 "//tensorflow/c/experimental/saved_model/core:concrete_function",
95 "//tensorflow/c/experimental/saved_model/core:concrete_function",
[all …]
|
| D | saved_model_api_test.cc | 77 TF_SavedModel* saved_model = in TEST_P() local
86 TF_DeleteSavedModel(saved_model); in TEST_P()
109 TF_SavedModel* saved_model = in TEST_P() local
114 TF_GetSavedModelConcreteFunction(saved_model, "compute", status); in TEST_P()
148 TF_DeleteSavedModel(saved_model); in TEST_P()
207 TF_SavedModel* saved_model = in TEST_P() local
212 TF_GetSavedModelSignatureDefFunction(saved_model, "serving_default", in TEST_P()
288 TF_DeleteSavedModel(saved_model); in TEST_P()
311 TF_SavedModel* saved_model = in TEST_P() local
316 TF_GetSavedModelConcreteFunction(saved_model, "read_file", status); in TEST_P()
[all …]
|
| D | saved_model_api.cc | 48 std::unique_ptr<tensorflow::TFSavedModelAPI> saved_model; in TF_LoadSavedModel() local
53 &saved_model); in TF_LoadSavedModel()
54 result = std::move(saved_model); in TF_LoadSavedModel()
78 std::unique_ptr<tensorflow::TFSavedModelAPI> saved_model; in TF_LoadSavedModelWithTags() local
83 &saved_model); in TF_LoadSavedModelWithTags()
84 result = std::move(saved_model); in TF_LoadSavedModelWithTags()
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | saved_model.py | 24 from tensorflow.python.saved_model import builder
25 from tensorflow.python.saved_model import constants
26 from tensorflow.python.saved_model import loader
27 from tensorflow.python.saved_model import main_op
28 from tensorflow.python.saved_model import method_name_updater
29 from tensorflow.python.saved_model import signature_constants
30 from tensorflow.python.saved_model import signature_def_utils
31 from tensorflow.python.saved_model import tag_constants
32 from tensorflow.python.saved_model import utils
33 from tensorflow.python.saved_model.load import load
[all …]
|
| D | signature_def_utils.py | 24 from tensorflow.python.saved_model.signature_def_utils_impl import build_signature_def
25 from tensorflow.python.saved_model.signature_def_utils_impl import classification_signature_def
26 from tensorflow.python.saved_model.signature_def_utils_impl import is_valid_signature
27 from tensorflow.python.saved_model.signature_def_utils_impl import load_op_from_signature_def
28 from tensorflow.python.saved_model.signature_def_utils_impl import op_signature_def
29 from tensorflow.python.saved_model.signature_def_utils_impl import predict_signature_def
30 from tensorflow.python.saved_model.signature_def_utils_impl import regression_signature_def
31 from tensorflow.python.saved_model.signature_def_utils_impl import supervised_eval_signature_def
32 from tensorflow.python.saved_model.signature_def_utils_impl import supervised_train_signature_def
|
| D | README.md | 15 * [Using the SavedModel Format](https://www.tensorflow.org/guide/saved_model)
22 ## [Public API](https://www.tensorflow.org/api_docs/python/tf/saved_model)
23 * [`tf.saved_model.save`](https://www.tensorflow.org/api_docs/python/tf/saved_model/save)
24 * [`tf.saved_model.load`](https://www.tensorflow.org/api_docs/python/tf/saved_model/load)
25 * [`tf.saved_model.SaveOptions`](https://www.tensorflow.org/api_docs/python/tf/saved_model/SaveOpti…
26 * [`tf.saved_model.LoadOptions`](https://www.tensorflow.org/api_docs/python/tf/saved_model/LoadOpti…
27 * [`tf.saved_model.Asset`](https://www.tensorflow.org/api_docs/python/tf/saved_model/Asset)
28 * [`tf.saved_model.contains_saved_model`](https://www.tensorflow.org/api_docs/python/tf/saved_model…
45 saved_model.pb
49 …* [`saved_model.pb`](https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/protob…
[all …]
|
| D | loader_impl.py | 34 from tensorflow.python.saved_model import constants
35 from tensorflow.python.saved_model import signature_def_utils
36 from tensorflow.python.saved_model import utils_impl as saved_model_utils
56 saved_model = _parse_saved_model(export_dir)
69 return (saved_model, debug_info)
95 saved_model = saved_model_pb2.SavedModel()
99 saved_model.ParseFromString(file_content)
100 return saved_model
106 text_format.Merge(file_content.decode("utf-8"), saved_model)
107 return saved_model
[all …]
|
| D | simple_save.py | 22 from tensorflow.python.saved_model import builder
23 from tensorflow.python.saved_model import signature_constants
24 from tensorflow.python.saved_model import signature_def_utils
25 from tensorflow.python.saved_model import tag_constants
|
| /external/tensorflow/tensorflow/python/distribute/integration_test/ |
| D | saved_model_test.py | 102 tf.saved_model.save(m, export_dir)
104 loaded = tf.saved_model.load(export_dir)
129 tf.saved_model.save(m, export_dir)
131 loaded = tf.saved_model.load(export_dir)
155 tf.saved_model.save(m, export_dir)
157 loaded = tf.saved_model.load(export_dir)
180 tf.saved_model.save(m, export_dir)
182 loaded = tf.saved_model.load(export_dir)
206 tf.saved_model.save(m, export_dir)
209 loaded = tf.saved_model.load(export_dir)
[all …]
|
| /external/tensorflow/tensorflow/python/saved_model/model_utils/ |
| D | __init__.py | 21 from tensorflow.python.saved_model.model_utils.export_output import *
22 from tensorflow.python.saved_model.model_utils.export_utils import build_all_signature_defs
23 from tensorflow.python.saved_model.model_utils.export_utils import export_outputs_for_mode
24 from tensorflow.python.saved_model.model_utils.export_utils import EXPORT_TAG_MAP
25 from tensorflow.python.saved_model.model_utils.export_utils import get_export_outputs
26 from tensorflow.python.saved_model.model_utils.export_utils import get_temp_export_dir
27 from tensorflow.python.saved_model.model_utils.export_utils import get_timestamped_export_dir
28 from tensorflow.python.saved_model.model_utils.export_utils import SIGNATURE_KEY_MAP
|
| D | BUILD | 52 "//tensorflow/python/saved_model:signature_def_utils",
75 "//tensorflow/python/saved_model:signature_constants",
88 "//tensorflow/python/saved_model:signature_constants",
89 "//tensorflow/python/saved_model:signature_def_utils",
90 "//tensorflow/python/saved_model:tag_constants",
110 "//tensorflow/python/saved_model:signature_constants",
111 "//tensorflow/python/saved_model:signature_def_utils",
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | BUILD | 6 # under tf/c/experimental/saved_model/public/. They are subject to change and
16 "//tensorflow/c/experimental/saved_model/internal:__pkg__",
67 "//tensorflow/c/experimental/saved_model/core/revived_types:asset",
68 "//tensorflow/c/experimental/saved_model/core/revived_types:constant",
69 "//tensorflow/c/experimental/saved_model/core/revived_types:partially_revived_objects",
70 … "//tensorflow/c/experimental/saved_model/core/revived_types:restored_resource_revival_state",
71 "//tensorflow/c/experimental/saved_model/core/revived_types:tf_concrete_function",
72 … "//tensorflow/c/experimental/saved_model/core/revived_types:tf_concrete_function_revival_state",
73 …"//tensorflow/c/experimental/saved_model/core/revived_types:tf_signature_def_function_revival_stat…
74 "//tensorflow/c/experimental/saved_model/core/revived_types:variable",
[all …]
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | BUILD | 55 "//tensorflow/python/saved_model:builder",
56 "//tensorflow/python/saved_model:load",
57 "//tensorflow/python/saved_model:loader",
58 "//tensorflow/python/saved_model:save",
59 "//tensorflow/python/saved_model:signature_constants",
60 "//tensorflow/python/saved_model:tag_constants",
81 "//tensorflow/python/saved_model:builder",
82 "//tensorflow/python/saved_model:loader",
83 "//tensorflow/python/saved_model:signature_constants",
84 "//tensorflow/python/saved_model:signature_def_utils",
[all …]
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | saved_model_utils.py | 27 from tensorflow.python.saved_model import constants
58 saved_model = saved_model_pb2.SavedModel()
62 saved_model.ParseFromString(file_content)
63 return saved_model
69 text_format.Merge(file_content.decode("utf-8"), saved_model)
70 return saved_model
89 saved_model = read_saved_model(saved_model_dir)
91 for meta_graph_def in saved_model.meta_graphs:
115 saved_model = read_saved_model(saved_model_dir)
118 for meta_graph_def in saved_model.meta_graphs:
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | BUILD | 33 "saved_model/base_serialization.py",
34 "saved_model/constants.py",
35 "saved_model/json_utils.py",
36 "saved_model/layer_serialization.py",
37 "saved_model/load.py",
38 "saved_model/metric_serialization.py",
39 "saved_model/model_serialization.py",
40 "saved_model/network_serialization.py",
41 "saved_model/save.py",
42 "saved_model/save_impl.py",
[all …]
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/ |
| D | README.md | 13 saved_model/
23 ## saved_model/public
25 `saved_model/public` is intended to house *only the public headers* of the
42 ## saved_model/internal
44 `saved_model/internal` is the "glue" between the C API and the internal C++
49 1. implement the C API functions declared in `saved_model/public`
51 2. define the C API types declared in `saved_model/public`
61 ## saved_model/core
63 `saved_model/core` contains pure C++ "Classes" underlying the C API types
64 in `saved_model/public/`. These are implementation
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/public/ |
| D | BUILD | 35 visibility = ["//tensorflow/c/experimental/saved_model/internal:__pkg__"],
59 actual = "//tensorflow/c/experimental/saved_model/internal:concrete_function",
64 actual = "//tensorflow/c/experimental/saved_model/internal:concrete_function_list",
69 actual = "//tensorflow/c/experimental/saved_model/internal:function_metadata",
74 actual = "//tensorflow/c/experimental/saved_model/internal:saved_model_api",
79 actual = "//tensorflow/c/experimental/saved_model/internal:signature_def_function",
84 actual = "//tensorflow/c/experimental/saved_model/internal:signature_def_function_metadata",
89 actual = "//tensorflow/c/experimental/saved_model/internal:signature_def_param",
94 actual = "//tensorflow/c/experimental/saved_model/internal:signature_def_param_list",
99 actual = "//tensorflow/c/experimental/saved_model/internal:tensor_spec",
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_saved_model_test.py | 35 from tensorflow.python.saved_model import saved_model
36 from tensorflow.python.saved_model import signature_constants
37 from tensorflow.python.saved_model import tag_constants
50 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
64 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
216 builder = saved_model.builder.SavedModelBuilder(saved_model_dir)
221 sig_input_tensor = saved_model.utils.build_tensor_info(in_tensor)
223 sig_output_tensor = saved_model.utils.build_tensor_info(out_tensor)
226 saved_model.signature_def_utils.build_signature_def(
228 saved_model.signature_constants.PREDICT_METHOD_NAME))
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/ |
| D | multi_arguments_results_v1.py | 61 tensor_info_x = tf.compat.v1.saved_model.utils.build_tensor_info(x)
62 tensor_info_y = tf.compat.v1.saved_model.utils.build_tensor_info(y)
63 tensor_info_s = tf.compat.v1.saved_model.utils.build_tensor_info(s)
64 tensor_info_t = tf.compat.v1.saved_model.utils.build_tensor_info(t)
67 'key': (tf.compat.v1.saved_model.signature_def_utils.build_signature_def(
77 'key2': (tf.compat.v1.saved_model.signature_def_utils.build_signature_def(
|
| D | shared_variable_v1.py | 51 tensor_info_x = tf.saved_model.utils.build_tensor_info(x)
52 tensor_info_r = tf.saved_model.utils.build_tensor_info(r)
54 signature_def = tf.saved_model.signature_def_utils.build_signature_def(
58 signature_def2 = tf.saved_model.signature_def_utils.build_signature_def(
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| D | BUILD | 27 "//tensorflow/python/saved_model:loader",
28 "//tensorflow/python/saved_model:signature_constants",
29 "//tensorflow/python/saved_model:tag_constants",
49 data = ["sample_model/saved_model.pb"],
58 "//tensorflow/python/saved_model:signature_constants",
59 "//tensorflow/python/saved_model:tag_constants",
|
| /external/tensorflow/tensorflow/cc/saved_model/experimental/public/ |
| D | BUILD | 22 "//tensorflow/c/experimental/saved_model/public:concrete_function",
34 "//tensorflow/c/experimental/saved_model/public:concrete_function_list",
44 "//tensorflow/c/experimental/saved_model/public:function_metadata",
57 "//tensorflow/c/experimental/saved_model/public:saved_model_api",
71 "//tensorflow/c/experimental/saved_model/public:signature_def_function",
82 "//tensorflow/c/experimental/saved_model/public:signature_def_function_metadata",
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | saved_model_save_load_test.py | 32 from tensorflow.python.saved_model import load_options as load_options_lib
33 from tensorflow.python.saved_model import save_options as save_options_lib
34 from tensorflow.python.saved_model import saved_model
46 saved_model.save(model, saved_dir)
90 func = saved_model.load(saved_dir)
122 saved_model.save(model, saved_dir, signatures=call)
130 model = saved_model.load(saved_dir)
174 saved_model.save(model, saved_dir, signatures=call, options=save_options)
179 loaded_model = saved_model.load(saved_dir, options=load_options)
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| D | BUILD | 8 "//tensorflow/c/experimental/saved_model/core:__pkg__",
26 "//tensorflow/cc/saved_model:constants",
95 "//tensorflow/c/experimental/saved_model/core:signature_def_function_metadata",
96 "//tensorflow/c/experimental/saved_model/core:tensor_spec",
165 "//tensorflow/c/experimental/saved_model/core/ops:variable_ops",
200 "//tensorflow/c/experimental/saved_model/core:concrete_function",
201 "//tensorflow/c/experimental/saved_model/core:function_metadata",
238 "//tensorflow/c/experimental/saved_model/core:signature_def_function",
239 "//tensorflow/c/experimental/saved_model/core:signature_def_function_metadata",
255 "//tensorflow/c/experimental/saved_model/core:signature_def_function_metadata",
|
| /external/tensorflow/tensorflow/python/tools/api/generator/ |
| D | api_init_files_v1.bzl | 72 "saved_model/__init__.py",
73 "saved_model/builder/__init__.py",
74 "saved_model/constants/__init__.py",
75 "saved_model/experimental/__init__.py",
76 "saved_model/loader/__init__.py",
77 "saved_model/main_op/__init__.py",
78 "saved_model/signature_constants/__init__.py",
79 "saved_model/signature_def_utils/__init__.py",
80 "saved_model/tag_constants/__init__.py",
81 "saved_model/utils/__init__.py",
|