Introduce `tf.data` namespace.

PiperOrigin-RevId: 170939033

1 files changed, 13 insertions, 10 deletions

diff --git a/tensorflow/python/__init__.py b/tensorflow/python/__init__.py
index 18603c2181..f3bdea92dd 100644
--- a/tensorflow/python/__init__.py
+++ b/tensorflow/python/__init__.py
@@ -78,9 +78,10 @@ from tensorflow.python.ops import linalg_ns as linalg
 # pylint: enable=wildcard-import
 
 # Bring in subpackages.
+from tensorflow.python import data
+from tensorflow.python import keras
 from tensorflow.python.estimator import estimator_lib as estimator
 from tensorflow.python.feature_column import feature_column_lib as feature_column
-from tensorflow.python import keras
 from tensorflow.python.layers import layers
 from tensorflow.python.ops import bitwise_ops as bitwise
 from tensorflow.python.ops import image_ops as image
@@ -91,10 +92,11 @@ from tensorflow.python.ops import spectral_ops as spectral
 from tensorflow.python.ops.distributions import distributions
 from tensorflow.python.ops.losses import losses
 from tensorflow.python.profiler import profiler
-from tensorflow.python.user_ops import user_ops
-from tensorflow.python.util import compat
 from tensorflow.python.saved_model import saved_model
 from tensorflow.python.summary import summary
+from tensorflow.python.user_ops import user_ops
+from tensorflow.python.util import compat
+
 
 # Import the names from python/training.py as train.Name.
 from tensorflow.python.training import training as train
@@ -222,6 +224,7 @@ _allowed_symbols.extend([
     'app',
     'bitwise',
     'compat',
+    'data',
     'distributions',
     'errors',
     'estimator',
@@ -231,12 +234,15 @@ _allowed_symbols.extend([
     'graph_util',
     'image',
     'initializers',
+    'keras',
+    'layers',
     'linalg',
     'logging',
     'losses',
     'metrics',
     'newaxis',
     'nn',
+    'profiler',
     'python_io',
     'resource_loader',
     'saved_model',
@@ -247,9 +253,6 @@ _allowed_symbols.extend([
     'test',
     'train',
     'user_ops',
-    'layers',
-    'profiler',
-    'keras',
 ])
 
 # Variables framework.versions:
@@ -263,11 +266,11 @@ _allowed_symbols.extend([
 # referenced in the whitelist.
 remove_undocumented(__name__, _allowed_symbols, [
     framework_lib, array_ops, check_ops, client_lib, compat, constant_op,
-    control_flow_ops, confusion_matrix_m, distributions,
-    functional_ops, histogram_ops, io_ops,
-    losses, math_ops, metrics, nn, resource_loader, sets, script_ops,
+    control_flow_ops, confusion_matrix_m, data, distributions,
+    functional_ops, histogram_ops, io_ops, keras, layers,
+    losses, math_ops, metrics, nn, profiler, resource_loader, sets, script_ops,
     session_ops, sparse_ops, state_ops, string_ops, summary, tensor_array_ops,
-    train, layers, profiler, keras
+    train
 ])
 
 # Special dunders that we choose to export: