Generalize milan datasets to support file formats other than TFRecords.
Generalize milan datasets to support file formats other than TFRecords.

PiperOrigin-RevId: 367236654

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo