[tf.data] Cache dataset structure properties in `ConcatenateDataset`.
Previously, these would be recomputed from the input on each call to `ConcatenateDataset.output_shapes`, which would lead to pathological behavior when building a long chain of `Dataset.concatenate()` transformations. PiperOrigin-RevId: 219217408
Loading
Please sign in to comment