diff options
author | 2018-02-09 11:16:01 -0800 | |
---|---|---|
committer | 2018-02-09 11:21:21 -0800 | |
commit | 6c85a66a16f07bab9b5dc3df33bc6b8111b76615 (patch) | |
tree | 53b3f02472af628d4b42904962bf99e9c1e2b837 /tensorflow/python/__init__.py | |
parent | 249b67204e5c1eb4805b8b021386aa0364045063 (diff) |
Add export calls for protos.
PiperOrigin-RevId: 185166764
Diffstat (limited to 'tensorflow/python/__init__.py')
-rw-r--r-- | tensorflow/python/__init__.py | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/tensorflow/python/__init__.py b/tensorflow/python/__init__.py index ea7604d30f..6d405d04b1 100644 --- a/tensorflow/python/__init__.py +++ b/tensorflow/python/__init__.py @@ -116,6 +116,7 @@ from tensorflow.python.platform import test from tensorflow.python.util.all_util import remove_undocumented from tensorflow.python.util.all_util import make_all +from tensorflow.python.util.tf_export import tf_export # Import modules whose docstrings contribute, for use by remove_undocumented # below. @@ -167,6 +168,31 @@ _allowed_symbols = [ 'TensorInfo', # Used for tf.saved_model functionality. ] +# Export protos +# pylint: disable=undefined-variable +tf_export('AttrValue')(AttrValue) +tf_export('ConfigProto')(ConfigProto) +tf_export('Event', 'summary.Event')(Event) +tf_export('GPUOptions')(GPUOptions) +tf_export('GraphDef')(GraphDef) +tf_export('GraphOptions')(GraphOptions) +tf_export('HistogramProto')(HistogramProto) +tf_export('LogMessage')(LogMessage) +tf_export('MetaGraphDef')(MetaGraphDef) +tf_export('NameAttrList')(NameAttrList) +tf_export('NodeDef')(NodeDef) +tf_export('OptimizerOptions')(OptimizerOptions) +tf_export('RunMetadata')(RunMetadata) +tf_export('RunOptions')(RunOptions) +tf_export('SessionLog', 'summary.SessionLog')(SessionLog) +tf_export('Summary', 'summary.Summary')(Summary) +tf_export('summary.SummaryDescription')(SummaryDescription) +tf_export('SummaryMetadata')(SummaryMetadata) +tf_export('summary.TaggedRunMetadata')(TaggedRunMetadata) +tf_export('TensorInfo')(TensorInfo) +# pylint: enable=undefined-variable + + # The following symbols are kept for compatibility. It is our plan # to remove them in the future. _allowed_symbols.extend([ |