Commit a63bbdbe authored by Derek Murray's avatar Derek Murray Committed by TensorFlower Gardener
Browse files

[tf.data] Add a `Structure._batch()` method for creating structures of batches. 

This change switches `Dataset.batch()`, `map_and_batch()`, and `sliding_window_batch()` to use the new helper. As a cascading side effect, all subclasses of `MapDataset` and `FlatMapDataset` are also switched to use `Structure` for representing their structure.

PiperOrigin-RevId: 223847841
parent c9e11180
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment