For perplexity_eval, if eval_dataset_fn is not specified, then fall back to train_dataset_fn.
For perplexity_eval, if eval_dataset_fn is not specified, then fall back to train_dataset_fn.

PiperOrigin-RevId: 296494834

tensorflow/mesh,tensorflow/mesh