Migrate pipelines to use dataset.map() instead of dataset.map_with_legacy_function().
Migrate pipelines to use dataset.map() instead of dataset.map_with_legacy_function().

PiperOrigin-RevId: 233884243

tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets