aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/estimator/exporter.py
diff options
context:
space:
mode:
authorGravatar Michael Case <mikecase@google.com>2018-06-07 12:05:24 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-06-07 12:14:50 -0700
commit501cf726cbee2ee13efef43884a6552ca211979d (patch)
tree2a93bae901b9f9d32f5d622e2e4d626668b48b99 /tensorflow/python/estimator/exporter.py
parent4d0d60a82c52c6c71650db33bf826f03559d91fc (diff)
Internal Change.
PiperOrigin-RevId: 199673803
Diffstat (limited to 'tensorflow/python/estimator/exporter.py')
-rw-r--r--tensorflow/python/estimator/exporter.py10
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.