Better performance with the tf.data API | TensorFlow Core