Update doc

PiperOrigin-RevId: 249559914

Author: Conchylicultor

docs/api_docs/python/_redirects.yaml

@@ -3,6 +3,8 @@ redirects:
   to: /datasets/api_docs/python/tfds/download/GenerateMode
 - from: /datasets/api_docs/python/tfds/testing/FeatureExpectationsTestCase/failureException
   to: /datasets/api_docs/python/tfds/testing/DatasetBuilderTestCase/failureException
+- from: /datasets/api_docs/python/tfds/testing/SubTestCase/failureException
+  to: /datasets/api_docs/python/tfds/testing/DatasetBuilderTestCase/failureException
 - from: /datasets/api_docs/python/tfds/testing/TestCase/failureException
   to: /datasets/api_docs/python/tfds/testing/DatasetBuilderTestCase/failureException
 - from: /datasets/api_docs/python/tfds/features/text

docs/api_docs/python/_toc.yaml

@@ -148,6 +148,8 @@ toc:
       path: /datasets/api_docs/python/tfds/testing/rm_tmp_dir
     - title: run_in_graph_and_eager_modes
       path: /datasets/api_docs/python/tfds/testing/run_in_graph_and_eager_modes
+    - title: SubTestCase
+      path: /datasets/api_docs/python/tfds/testing/SubTestCase
     - title: TestCase
       path: /datasets/api_docs/python/tfds/testing/TestCase
     - title: test_main

docs/api_docs/python/index.md

@@ -65,6 +65,8 @@
 *   <a href="./tfds/testing/FeatureExpectationItem.md"><code>tfds.testing.FeatureExpectationItem</code></a>
 *   <a href="./tfds/testing/FeatureExpectationsTestCase.md"><code>tfds.testing.FeatureExpectationsTestCase</code></a>
 *   <a href="./tfds/testing/DatasetBuilderTestCase/failureException.md"><code>tfds.testing.FeatureExpectationsTestCase.failureException</code></a>
+*   <a href="./tfds/testing/SubTestCase.md"><code>tfds.testing.SubTestCase</code></a>
+*   <a href="./tfds/testing/DatasetBuilderTestCase/failureException.md"><code>tfds.testing.SubTestCase.failureException</code></a>
 *   <a href="./tfds/testing/TestCase.md"><code>tfds.testing.TestCase</code></a>
 *   <a href="./tfds/testing/DatasetBuilderTestCase/failureException.md"><code>tfds.testing.TestCase.failureException</code></a>
 *   <a href="./tfds/testing/make_tmp_dir.md"><code>tfds.testing.make_tmp_dir</code></a>

docs/api_docs/python/tfds/_api_cache.json

@@ -30,6 +30,5 @@ and `tf.Tensor`s to iterables of NumPy arrays and NumPy arrays, respectively.
 *   <b>`graph`</b>: `tf.Graph`, optional, explicitly set the graph to use.
 
 #### Returns:
-
 A structure matching `dataset` where `tf.data.Dataset`s are converted to
 generators of NumPy arrays and `tf.Tensor`s are converted to NumPy arrays.

docs/api_docs/python/tfds/as_numpy.md

@@ -72,12 +72,13 @@ Callers must pass arguments as keyword arguments.
 
 <h3 id="as_dataset"><code>as_dataset</code></h3>
 
-``` python
+```python
 as_dataset(
     split=None,
     batch_size=1,
     shuffle_files=None,
-    as_supervised=False
+    as_supervised=False,
+    in_memory=None
 )
 ```
 
@@ -101,6 +102,11 @@ Callers must pass arguments as keyword arguments.
     will have a 2-tuple structure `(input, label)` according to
     `builder.info.supervised_keys`. If `False`, the default, the returned
     `tf.data.Dataset` will have a dictionary with all the features.
+*   <b>`in_memory`</b>: `bool`, if `True`, loads the dataset in memory which
+    increases iteration speeds. Note that if `True` and the dataset has unknown
+    dimensions, the features will be padded to the maximum size across the
+    dataset. By default (when `None`), will load the dataset in memory if the
+    size is <1GB and all feature dimensions are statically known.
 
 #### Returns:
 `tf.data.Dataset`, or if `split=None`, `dict<key: tfds.Split, value:

docs/api_docs/python/tfds/core/BeamBasedBuilder.md

@@ -103,12 +103,13 @@ Callers must pass arguments as keyword arguments.
 
 <h3 id="as_dataset"><code>as_dataset</code></h3>
 
-``` python
+```python
 as_dataset(
     split=None,
     batch_size=1,
     shuffle_files=None,
-    as_supervised=False
+    as_supervised=False,
+    in_memory=None
 )
 ```
 
@@ -132,6 +133,11 @@ Callers must pass arguments as keyword arguments.
     will have a 2-tuple structure `(input, label)` according to
     `builder.info.supervised_keys`. If `False`, the default, the returned
     `tf.data.Dataset` will have a dictionary with all the features.
+*   <b>`in_memory`</b>: `bool`, if `True`, loads the dataset in memory which
+    increases iteration speeds. Note that if `True` and the dataset has unknown
+    dimensions, the features will be padded to the maximum size across the
+    dataset. By default (when `None`), will load the dataset in memory if the
+    size is <1GB and all feature dimensions are statically known.
 
 #### Returns:
 `tf.data.Dataset`, or if `split=None`, `dict<key: tfds.Split, value:

docs/api_docs/python/tfds/core/DatasetBuilder.md

@@ -81,12 +81,13 @@ Callers must pass arguments as keyword arguments.
 
 <h3 id="as_dataset"><code>as_dataset</code></h3>
 
-``` python
+```python
 as_dataset(
     split=None,
     batch_size=1,
     shuffle_files=None,
-    as_supervised=False
+    as_supervised=False,
+    in_memory=None
 )
 ```
 
@@ -110,6 +111,11 @@ Callers must pass arguments as keyword arguments.
     will have a 2-tuple structure `(input, label)` according to
     `builder.info.supervised_keys`. If `False`, the default, the returned
     `tf.data.Dataset` will have a dictionary with all the features.
+*   <b>`in_memory`</b>: `bool`, if `True`, loads the dataset in memory which
+    increases iteration speeds. Note that if `True` and the dataset has unknown
+    dimensions, the features will be padded to the maximum size across the
+    dataset. By default (when `None`), will load the dataset in memory if the
+    size is <1GB and all feature dimensions are statically known.
 
 #### Returns:
 `tf.data.Dataset`, or if `split=None`, `dict<key: tfds.Split, value:

docs/api_docs/python/tfds/core/GeneratorBasedBuilder.md

@@ -32,7 +32,6 @@ The resulting split will correspond to 25% of the train split merged with
 100% of the test split.
 
 #### Warning:
-
 A split cannot be added twice, so the following will fail:
 
 ```
@@ -44,7 +43,6 @@ split = tfds.Split.TEST + tfds.Split.ALL  # Error
 ```
 
 #### Warning:
-
 The slices can be applied only one time. So the following are valid:
 
 ```

docs/api_docs/python/tfds/core/NamedSplit.md

@@ -33,10 +33,11 @@ Defined in [`core/features/__init__.py`](https://github.com/tensorflow/datasets/
 
 [`class Tensor`](../tfds/features/Tensor.md): `FeatureConnector` for generic data of arbitrary shape and type.
 
-[`class TensorInfo`](../tfds/features/TensorInfo.md): TensorInfo(shape, dtype)
+[`class TensorInfo`](../tfds/features/TensorInfo.md): Structure containing info
+on the `tf.Tensor` shape/dtype.
 
 [`class Sequence`](../tfds/features/Sequence.md): Composite `FeatureConnector`
-where each value is a list.
+for a `dict` where each value is a list.
 
 [`class Image`](../tfds/features/Image.md): `FeatureConnector` for images.