diff options
| author |  Karmel Allison <karmel@google.com> | 2018-05-04 16:01:02 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-05 08:30:01 -0700 |
| commit | 008a3b69a601dc68fd940eb8a03b0c445714a339 (patch) | |
| tree | df7a92de37594adc3d8a3aef72baea1ea137fb1c /tensorflow/python/saved_model/signature_constants.py | |
| parent | ab48fb528221152299fb08da8116d2eca54b8423 (diff) | |
Add the ability to export separate SavedModels for train and eval mode to Estimator with two new methods, available in tf.contrib: export_all_saved_models and export_saved_model_for_mode.
PiperOrigin-RevId: 195485922
Diffstat (limited to 'tensorflow/python/saved_model/signature_constants.py')
| -rw-r--r-- | tensorflow/python/saved_model/signature_constants.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/python/saved_model/signature_constants.py b/tensorflow/python/saved_model/signature_constants.py index 819f351291..99007a9634 100644 --- a/tensorflow/python/saved_model/signature_constants.py +++ b/tensorflow/python/saved_model/signature_constants.py @@ -94,3 +94,9 @@ tf_export("saved_model.signature_constants.REGRESS_OUTPUTS").export_constant( __name__, "REGRESS_OUTPUTS") ################################################################################ +# Train/Eval API constants. +# Not exported while export_all_saved_models is in contrib. + +SUPERVISED_TRAIN_METHOD_NAME = "tensorflow/supervised/training" + +SUPERVISED_EVAL_METHOD_NAME = "tensorflow/supervised/eval" |