Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/tensorflow/python/saved_model/simple_save.py: 73%
15 statements
« prev ^ index » next coverage.py v7.4.0, created at 2024-01-03 07:57 +0000
« prev ^ index » next coverage.py v7.4.0, created at 2024-01-03 07:57 +0000
1# Copyright 2017 The TensorFlow Authors. All Rights Reserved.
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7# http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14# ==============================================================================
15"""SavedModel simple save functionality."""
17from tensorflow.python.framework import ops
18from tensorflow.python.saved_model import builder
19from tensorflow.python.saved_model import signature_constants
20from tensorflow.python.saved_model import signature_def_utils
21from tensorflow.python.saved_model import tag_constants
22from tensorflow.python.util import deprecation
23from tensorflow.python.util.tf_export import tf_export
26@tf_export(v1=['saved_model.simple_save'])
27@deprecation.deprecated(
28 None,
29 'This API was designed for TensorFlow v1. See https://www.tensorflow.org/guide/migrate '
30 'for instructions on how to migrate your code to TensorFlow v2.')
31def simple_save(session, export_dir, inputs, outputs, legacy_init_op=None):
32 """Convenience function to build a SavedModel suitable for serving.
34 In many common cases, saving models for serving will be as simple as:
36 simple_save(session,
37 export_dir,
38 inputs={"x": x, "y": y},
39 outputs={"z": z})
41 Although in many cases it's not necessary to understand all of the many ways
42 to configure a SavedModel, this method has a few practical implications:
43 - It will be treated as a graph for inference / serving (i.e. uses the tag
44 `saved_model.SERVING`)
45 - The SavedModel will load in TensorFlow Serving and supports the
46 [Predict
47 API](https://github.com/tensorflow/serving/blob/master/tensorflow_serving/apis/predict.proto).
48 To use the Classify, Regress, or MultiInference APIs, please
49 use either
50 [tf.Estimator](https://www.tensorflow.org/api_docs/python/tf/estimator/Estimator)
51 or the lower level
52 [SavedModel
53 APIs](https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/saved_model/README.md).
54 - Some TensorFlow ops depend on information on disk or other information
55 called "assets". These are generally handled automatically by adding the
56 assets to the `GraphKeys.ASSET_FILEPATHS` collection. Only assets in that
57 collection are exported; if you need more custom behavior, you'll need to
58 use the
59 [SavedModelBuilder](https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/saved_model/builder.py).
61 More information about SavedModel and signatures can be found here:
62 https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/saved_model/README.md.
64 Args:
65 session: The TensorFlow session from which to save the meta graph and
66 variables.
67 export_dir: The path to which the SavedModel will be stored.
68 inputs: dict mapping string input names to tensors. These are added
69 to the SignatureDef as the inputs.
70 outputs: dict mapping string output names to tensors. These are added
71 to the SignatureDef as the outputs.
72 legacy_init_op: Legacy support for op or group of ops to execute after the
73 restore op upon a load.
74 """
75 signature_def_map = {
76 signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY:
77 signature_def_utils.predict_signature_def(inputs, outputs)
78 }
79 b = builder.SavedModelBuilder(export_dir)
80 b.add_meta_graph_and_variables(
81 session,
82 tags=[tag_constants.SERVING],
83 signature_def_map=signature_def_map,
84 assets_collection=ops.get_collection(ops.GraphKeys.ASSET_FILEPATHS),
85 main_op=legacy_init_op,
86 clear_devices=True)
87 b.save()