Commit 93d5185f authored by Rachel Lim's avatar Rachel Lim Committed by TensorFlower Gardener
Browse files

[tf.data] Make map_vectorization optimization work with... 

[tf.data] Make map_vectorization optimization work with ExperimentalMapAndBatchDataset as well. i.e. dataset.apply(tf.data.experimental.map_and_batch(map_fn, ...)) -> dataset.batch(...).map(vectorized_map_fn)

PiperOrigin-RevId: 230422450
parent 522066c8
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