Add the ability to export separate SavedModels for train and eval mode to Estimator...
authorKarmel Allison <karmel@google.com>
Fri, 4 May 2018 23:01:02 +0000 (16:01 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Sat, 5 May 2018 15:30:01 +0000 (08:30 -0700)
commit008a3b69a601dc68fd940eb8a03b0c445714a339
treedf7a92de37594adc3d8a3aef72baea1ea137fb1c
parentab48fb528221152299fb08da8116d2eca54b8423
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
20 files changed:
tensorflow/contrib/estimator/BUILD
tensorflow/contrib/estimator/__init__.py
tensorflow/contrib/estimator/python/estimator/export.py [new file with mode: 0644]
tensorflow/contrib/estimator/python/estimator/export_test.py [new file with mode: 0644]
tensorflow/python/estimator/BUILD
tensorflow/python/estimator/estimator.py
tensorflow/python/estimator/estimator_test.py
tensorflow/python/estimator/export/export.py
tensorflow/python/estimator/export/export_output.py
tensorflow/python/estimator/export/export_output_test.py
tensorflow/python/estimator/export/export_test.py
tensorflow/python/estimator/model_fn.py
tensorflow/python/saved_model/builder_impl.py
tensorflow/python/saved_model/constants.py
tensorflow/python/saved_model/saved_model_test.py
tensorflow/python/saved_model/signature_constants.py
tensorflow/python/saved_model/signature_def_utils.py
tensorflow/python/saved_model/signature_def_utils_impl.py
tensorflow/python/saved_model/signature_def_utils_test.py
tensorflow/python/saved_model/tag_constants.py