message: Set default dataset options when building dataset with shuffle.
  It requires stateful op under the hood, which might raise error in TPU.
diff: @@ -336,6 +336,9 @@ class TensorLikeDataAdapter(DataAdapter):
  # input pipeline graph serialization and deserialization
  options = dataset_ops.Options()
  options.experimental_optimization.apply_default_optimizations = False
  + if shuffle:
  + # See b/141490660 for more details.
  + options.experimental_allow_stateful = True
  dataset = dataset.with_options(options)
  self._dataset = dataset
  