tree 4482d4671006f8a57dc69128b2882bbcd9057006
parent b945234aa97d3fd8d3a992ead20b33b6fdde04a2
author Rachel Lim <rachelim@google.com> 1568237771 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1568238533 -0700

[tf.data] Add a new `RebatchDatasetV2` op that does rebatching (instead of
rebatching via graph rewrites). Also adds a `prefetch` after `RebatchDataset` for performance.

PiperOrigin-RevId: 268544896