Commit cca3a5ab authored by Piotr Padlewski's avatar Piotr Padlewski Committed by TensorFlower Gardener
Browse files

[tf.data] Adding optimization for fusing consecutive `map(...)` transformations.

This optimization pass fuses maps together e.g:

  map(x: x*x).map(x: x+1)

can be fused to:
  map(x: x*x+1)

We can see 10-x speedup on simple pipeline containing 50 consecutive maps of identity.

PiperOrigin-RevId: 206069355
parent 89bcc80f
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment