[Cloud TPU / Keras]: Pipeline inputs for performance.
With this pipelining change (and a little bit of re-tuning of input pipelines to have _less_ parallelism to avoid thread starvation), we are able to significantly reduce the overheads of supporting dynamic shapes with Keras. PiperOrigin-RevId: 211157531
Loading
Please sign in to comment