| /external/tensorflow/tensorflow/core/tfrt/saved_model/tests/ |
| D | BUILD | 16 "//tensorflow/core/tfrt/saved_model/tests/...",
43 "//tensorflow/python/saved_model:builder",
44 "//tensorflow/python/saved_model:save",
45 "//tensorflow/python/saved_model:save_options",
46 "//tensorflow/python/saved_model:signature_constants",
47 "//tensorflow/python/saved_model:signature_def_utils",
48 "//tensorflow/python/saved_model:tag_constants",
49 "//tensorflow/python/saved_model:utils",
66 "//tensorflow/python/saved_model:builder",
67 "//tensorflow/python/saved_model:signature_constants",
[all …]
|
| D | saved_model_test.cc | 63 auto saved_model = in TEST_P() local
76 TF_ASSERT_OK(saved_model->Run(run_options, "toy", inputs, &outputs)); in TEST_P()
108 auto saved_model = in TEST() local
122 TF_ASSERT_OK(saved_model->Run(run_options, "toy", inputs, &outputs)); in TEST()
128 auto op_count = saved_model->GetHostContext() in TEST()
177 auto saved_model = in TEST() local
192 TF_ASSERT_OK(saved_model->Run({}, "serving_default", inputs, &outputs)); in TEST()
225 auto saved_model = in TEST() local
250 TF_ASSERT_OK(saved_model->RunMultipleSignatures(/*run_options=*/{}, names, in TEST()
256 auto toy_metadata = saved_model->GetFunctionMetadata("toy"); in TEST()
[all …]
|
| D | gen_error_v1.py | 23 from tensorflow.python.saved_model import builder
24 from tensorflow.python.saved_model import signature_constants
25 from tensorflow.python.saved_model import signature_def_utils
26 from tensorflow.python.saved_model import tag_constants
27 from tensorflow.python.saved_model import utils
|
| D | gen_pow.py | 24 from tensorflow.python.saved_model import builder
25 from tensorflow.python.saved_model import signature_constants
26 from tensorflow.python.saved_model import signature_def_utils
27 from tensorflow.python.saved_model import tag_constants
28 from tensorflow.python.saved_model import utils
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| D | BUILD | 10 # C API functions exposed in tf/c/experimental/saved_model/public/.
37 "//tensorflow/c/experimental/saved_model/public:concrete_function.h",
41 "//tensorflow/c/experimental/saved_model/public:__pkg__",
54 "//tensorflow/c/experimental/saved_model/core:concrete_function",
55 "//tensorflow/c/experimental/saved_model/core:function_metadata",
67 "//tensorflow/c/experimental/saved_model/public:concrete_function_list.h",
71 "//tensorflow/c/experimental/saved_model/public:__pkg__",
78 "//tensorflow/c/experimental/saved_model/core:concrete_function",
88 "//tensorflow/c/experimental/saved_model/core:concrete_function",
99 "//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 | 20 from tensorflow.python.saved_model import builder
21 from tensorflow.python.saved_model import constants
22 from tensorflow.python.saved_model import loader
23 from tensorflow.python.saved_model import main_op
24 from tensorflow.python.saved_model import method_name_updater
25 from tensorflow.python.saved_model import signature_constants
26 from tensorflow.python.saved_model import signature_def_utils
27 from tensorflow.python.saved_model import tag_constants
28 from tensorflow.python.saved_model import utils
29 from tensorflow.python.saved_model.load import load
[all …]
|
| D | signature_def_utils.py | 20 from tensorflow.python.saved_model.signature_def_utils_impl import build_signature_def
21 from tensorflow.python.saved_model.signature_def_utils_impl import classification_signature_def
22 from tensorflow.python.saved_model.signature_def_utils_impl import is_valid_signature
23 from tensorflow.python.saved_model.signature_def_utils_impl import load_op_from_signature_def
24 from tensorflow.python.saved_model.signature_def_utils_impl import op_signature_def
25 from tensorflow.python.saved_model.signature_def_utils_impl import predict_signature_def
26 from tensorflow.python.saved_model.signature_def_utils_impl import regression_signature_def
27 from tensorflow.python.saved_model.signature_def_utils_impl import supervised_eval_signature_def
28 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 | 31 from tensorflow.python.saved_model import constants
32 from tensorflow.python.saved_model import signature_def_utils
33 from tensorflow.python.saved_model import utils_impl as saved_model_utils
34 from tensorflow.python.saved_model.pywrap_saved_model import metrics
57 saved_model = parse_saved_model(export_dir)
70 return (saved_model, debug_info)
97 saved_model = saved_model_pb2.SavedModel()
102 saved_model.ParseFromString(file_content)
103 return saved_model
110 text_format.Merge(file_content.decode("utf-8"), saved_model)
[all …]
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | BUILD | 51 "//tensorflow/python/saved_model:builder",
52 "//tensorflow/python/saved_model:load",
53 "//tensorflow/python/saved_model:loader",
54 "//tensorflow/python/saved_model:save",
55 "//tensorflow/python/saved_model:signature_constants",
56 "//tensorflow/python/saved_model:tag_constants",
68 "//tensorflow/python/saved_model:builder",
69 "//tensorflow/python/saved_model:loader",
70 "//tensorflow/python/saved_model:signature_constants",
71 "//tensorflow/python/saved_model:signature_def_utils",
[all …]
|
| /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__",
51 "//tensorflow/cc/saved_model:bundle_v2",
69 "//tensorflow/c/experimental/saved_model/core/revived_types:asset",
70 "//tensorflow/c/experimental/saved_model/core/revived_types:constant",
71 "//tensorflow/c/experimental/saved_model/core/revived_types:partially_revived_objects",
72 … "//tensorflow/c/experimental/saved_model/core/revived_types:restored_resource_revival_state",
73 "//tensorflow/c/experimental/saved_model/core/revived_types:tf_concrete_function",
74 … "//tensorflow/c/experimental/saved_model/core/revived_types:tf_concrete_function_revival_state",
75 …"//tensorflow/c/experimental/saved_model/core/revived_types:tf_signature_def_function_revival_stat…
[all …]
|
| /external/tensorflow/tensorflow/python/saved_model/registration/ |
| D | __init__.py | 17 from tensorflow.python.saved_model.registration.registration import get_registered_class
18 from tensorflow.python.saved_model.registration.registration import get_registered_class_name
19 from tensorflow.python.saved_model.registration.registration import get_registered_saver_name
20 from tensorflow.python.saved_model.registration.registration import get_restore_function
21 from tensorflow.python.saved_model.registration.registration import get_save_function
22 from tensorflow.python.saved_model.registration.registration import get_strict_predicate_restore
25 from tensorflow.python.saved_model.registration.registration import register_checkpoint_saver
26 from tensorflow.python.saved_model.registration.registration import register_serializable
28 from tensorflow.python.saved_model.registration.registration import RegisteredSaver
29 from tensorflow.python.saved_model.registration.registration import validate_restore_function
|
| /external/tensorflow/tensorflow/python/distribute/integration_test/ |
| D | saved_model_test.py | 95 tf.saved_model.save(m, export_dir)
152 tf.saved_model.save(m, export_dir)
154 loaded = tf.saved_model.load(export_dir)
179 tf.saved_model.save(m, export_dir)
181 loaded = tf.saved_model.load(export_dir)
205 tf.saved_model.save(m, export_dir)
207 loaded = tf.saved_model.load(export_dir)
230 tf.saved_model.save(m, export_dir)
232 loaded = tf.saved_model.load(export_dir)
256 tf.saved_model.save(m, export_dir)
[all …]
|
| /external/tensorflow/tensorflow/python/saved_model/model_utils/ |
| D | __init__.py | 18 from tensorflow.python.saved_model.model_utils.export_output import *
19 from tensorflow.python.saved_model.model_utils.export_utils import build_all_signature_defs
20 from tensorflow.python.saved_model.model_utils.export_utils import export_outputs_for_mode
21 from tensorflow.python.saved_model.model_utils.export_utils import EXPORT_TAG_MAP
22 from tensorflow.python.saved_model.model_utils.export_utils import get_export_outputs
23 from tensorflow.python.saved_model.model_utils.export_utils import get_temp_export_dir
24 from tensorflow.python.saved_model.model_utils.export_utils import get_timestamped_export_dir
25 from tensorflow.python.saved_model.model_utils.export_utils import SIGNATURE_KEY_MAP
|
| D | BUILD | 50 "//tensorflow/python/saved_model:signature_def_utils",
72 "//tensorflow/python/saved_model:signature_constants",
87 "//tensorflow/python/saved_model:signature_constants",
88 "//tensorflow/python/saved_model:signature_def_utils",
89 "//tensorflow/python/saved_model:tag_constants",
90 "//tensorflow/python/saved_model:utils",
110 "//tensorflow/python/saved_model:signature_constants",
111 "//tensorflow/python/saved_model:signature_def_utils",
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | saved_model_utils.py | 23 from tensorflow.python.saved_model import constants
54 saved_model = saved_model_pb2.SavedModel()
59 saved_model.ParseFromString(file_content)
60 return saved_model
67 text_format.Merge(file_content.decode("utf-8"), saved_model)
68 return saved_model
87 saved_model = read_saved_model(saved_model_dir)
89 for meta_graph_def in saved_model.meta_graphs:
113 saved_model = read_saved_model(saved_model_dir)
118 for meta_graph_def in saved_model.meta_graphs:
|
| /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/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/lite/python/ |
| D | convert_saved_model_test.py | 31 from tensorflow.python.saved_model import saved_model
32 from tensorflow.python.saved_model import signature_constants
33 from tensorflow.python.saved_model import tag_constants
46 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
60 saved_model.simple_save(sess, saved_model_dir, inputs, outputs)
212 builder = saved_model.builder.SavedModelBuilder(saved_model_dir)
217 sig_input_tensor = saved_model.utils.build_tensor_info(in_tensor)
219 sig_output_tensor = saved_model.utils.build_tensor_info(out_tensor)
222 saved_model.signature_def_utils.build_signature_def(
224 saved_model.signature_constants.PREDICT_METHOD_NAME))
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/ |
| D | multi_arguments_results_v1.py | 57 tensor_info_x = tf.compat.v1.saved_model.utils.build_tensor_info(x)
58 tensor_info_y = tf.compat.v1.saved_model.utils.build_tensor_info(y)
59 tensor_info_s = tf.compat.v1.saved_model.utils.build_tensor_info(s)
60 tensor_info_t = tf.compat.v1.saved_model.utils.build_tensor_info(t)
63 'key': (tf.compat.v1.saved_model.signature_def_utils.build_signature_def(
73 'key2': (tf.compat.v1.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",
48 data = ["sample_model/saved_model.pb"],
59 "//tensorflow/python/saved_model:signature_constants",
60 "//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/cc/saved_model/ |
| D | util.cc | 23 namespace saved_model { namespace
25 std::string GetWriteVersion(const SavedModel& saved_model) { in GetWriteVersion() argument
26 if (saved_model.meta_graphs_size() == 1 && in GetWriteVersion()
27 saved_model.meta_graphs()[0].has_object_graph_def()) { in GetWriteVersion()
|