diff options
author | Michael Case <mikecase@google.com> | 2018-08-21 12:09:06 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-21 12:12:54 -0700 |
commit | 041b02c3b48a9b0cb085ca02f8674171f5d31fca (patch) | |
tree | 3ab1cf93a6399108789e6916f1dc7922799b4835 | |
parent | 81d90de884ad6005e57e0d7d333e8476659d00c2 (diff) |
Internal Change.
PiperOrigin-RevId: 209637025
-rw-r--r-- | tensorflow/python/BUILD | 1 | ||||
-rw-r--r-- | tensorflow/python/tools/api/generator/BUILD | 5 | ||||
-rw-r--r-- | tensorflow/python/training/warm_starting_util.py | 2 | ||||
-rw-r--r-- | tensorflow/python/util/tf_export.py | 2 |
4 files changed, 5 insertions, 5 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index e6d78301a5..f055b56809 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -143,6 +143,7 @@ py_library( "//tensorflow/python/profiler", "//tensorflow/python/saved_model", "//tensorflow/python/tools:component_api_helper", + "//tensorflow/python/tools/api/generator:create_python_api", "//third_party/py/numpy", ], ) diff --git a/tensorflow/python/tools/api/generator/BUILD b/tensorflow/python/tools/api/generator/BUILD index f87fdb2d88..36af091163 100644 --- a/tensorflow/python/tools/api/generator/BUILD +++ b/tensorflow/python/tools/api/generator/BUILD @@ -14,14 +14,13 @@ exports_files( ], ) -py_binary( +py_library( name = "create_python_api", srcs = ["//tensorflow/python/tools/api/generator:create_python_api.py"], - main = "//tensorflow/python/tools/api/generator:create_python_api.py", srcs_version = "PY2AND3", visibility = ["//visibility:public"], deps = [ - "//tensorflow/python:no_contrib", + "//tensorflow/python:util", "//tensorflow/python/tools/api/generator:doc_srcs", ], ) diff --git a/tensorflow/python/training/warm_starting_util.py b/tensorflow/python/training/warm_starting_util.py index 0ba7ba983d..c0dd46bfa5 100644 --- a/tensorflow/python/training/warm_starting_util.py +++ b/tensorflow/python/training/warm_starting_util.py @@ -32,7 +32,7 @@ from tensorflow.python.training import saver from tensorflow.python.util.tf_export import tf_export -@tf_export("train.VocabInfo", allow_multiple_exports=True) +@tf_export("train.VocabInfo") class VocabInfo( collections.namedtuple("VocabInfo", [ "new_vocab", diff --git a/tensorflow/python/util/tf_export.py b/tensorflow/python/util/tf_export.py index 274f32c21f..2be4dbb283 100644 --- a/tensorflow/python/util/tf_export.py +++ b/tensorflow/python/util/tf_export.py @@ -213,4 +213,4 @@ class api_export(object): # pylint: disable=invalid-name tf_export = functools.partial(api_export, api_name=TENSORFLOW_API_NAME) -estimator_export = functools.partial(tf_export, api_name=ESTIMATOR_API_NAME) +estimator_export = functools.partial(api_export, api_name=ESTIMATOR_API_NAME) |