tf.compat.v1.data.experimental.make_batched_features_dataset

View source on GitHub