Adding tf.data optimization for rewriting `map(...).batch(...)` to `map_and_batch...
authorJiri Simsa <jsimsa@google.com>
Mon, 28 May 2018 16:33:49 +0000 (09:33 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Mon, 28 May 2018 16:36:27 +0000 (09:36 -0700)
commitb05a6b5c4cb685b19b8c09693d40d4743af79dea
treebc8ae69747637306ab76f15a55473578a3559db4
parent2d83e131b930581b15a50538a020bda30af08ad4
Adding tf.data optimization for rewriting `map(...).batch(...)` to `map_and_batch(...)`.

PiperOrigin-RevId: 198310806
tensorflow/core/grappler/optimizers/data/BUILD
tensorflow/core/grappler/optimizers/data/map_and_batch_fusion.cc [new file with mode: 0644]
tensorflow/core/grappler/optimizers/data/map_and_batch_fusion.h [new file with mode: 0644]
tensorflow/core/grappler/optimizers/data/map_and_batch_fusion_test.cc [new file with mode: 0644]