message: [tf.data] Remove element_spec from dataset saving example code.
  element_spec is no longer needed.
diff: @@ -49,8 +49,7 @@ def save(dataset, path, compression=None, shard_func=None):
  >>> # Save a dataset
  >>> dataset = tf.data.Dataset.range(2)
  >>> tf.data.experimental.save(dataset, path)
  - >>> new_dataset = tf.data.experimental.load(path,
  - ... tf.TensorSpec(shape=(), dtype=tf.int64))
  + >>> new_dataset = tf.data.experimental.load(path)
  >>> for elem in new_dataset:
  ... print(elem)
  tf.Tensor(0, shape=(), dtype=int64)
  