Fixed typo referencing tf.Dataset
Fixed typo referencing tf.Dataset

PiperOrigin-RevId: 244926626

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor