Make decode_from_file work on TPU. For Mesh-TensorFlow, this will only work after the fix to avoid concatentating the outfeeds in BROADCAST mode.
Make decode_from_file work on TPU.
For Mesh-TensorFlow, this will only work after the fix to avoid concatentating the outfeeds in BROADCAST mode.

PiperOrigin-RevId: 226048381

tensorflow/mesh,tensorflow/mesh