diff options
author | 2018-06-07 12:05:24 -0700 | |
---|---|---|
committer | 2018-06-07 12:14:50 -0700 | |
commit | 501cf726cbee2ee13efef43884a6552ca211979d (patch) | |
tree | 2a93bae901b9f9d32f5d622e2e4d626668b48b99 /tensorflow/python/estimator/exporter.py | |
parent | 4d0d60a82c52c6c71650db33bf826f03559d91fc (diff) |
Internal Change.
PiperOrigin-RevId: 199673803
Diffstat (limited to 'tensorflow/python/estimator/exporter.py')
-rw-r--r-- | tensorflow/python/estimator/exporter.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/tensorflow/python/estimator/exporter.py b/tensorflow/python/estimator/exporter.py index 5981fa59b7..7cdf840c97 100644 --- a/tensorflow/python/estimator/exporter.py +++ b/tensorflow/python/estimator/exporter.py @@ -28,10 +28,10 @@ from tensorflow.python.framework import errors_impl from tensorflow.python.platform import gfile from tensorflow.python.platform import tf_logging from tensorflow.python.summary import summary_iterator -from tensorflow.python.util.tf_export import tf_export +from tensorflow.python.util.tf_export import estimator_export -@tf_export('estimator.Exporter') +@estimator_export('estimator.Exporter') class Exporter(object): """A class representing a type of model export.""" @@ -172,7 +172,7 @@ def _verify_compre_fn_args(compare_fn): (compare_fn, non_valid_args)) -@tf_export('estimator.BestExporter') +@estimator_export('estimator.BestExporter') class BestExporter(Exporter): """This class exports the serving graph and checkpoints of the best models. @@ -367,7 +367,7 @@ class BestExporter(Exporter): return best_eval_result -@tf_export('estimator.FinalExporter') +@estimator_export('estimator.FinalExporter') class FinalExporter(Exporter): """This class exports the serving graph and checkpoints in the end. @@ -418,7 +418,7 @@ class FinalExporter(Exporter): is_the_final_export) -@tf_export('estimator.LatestExporter') +@estimator_export('estimator.LatestExporter') class LatestExporter(Exporter): """This class regularly exports the serving graph and checkpoints. |