message: Update the docstring for `infer_steps_for_dataset`.
diff: @@ -1597,8 +1597,9 @@ def infer_steps_for_dataset(dataset, steps, epochs=1, steps_name='steps'):
  Returns:
  Integer or `None`. Inferred number of steps to loop through the dataset.
  - `None` is returned if the size of the dataset is unknown and `steps` was
  - not specified.
  + `None` is returned if 1) the size of the dataset is unknown and `steps` was
  + not specified, or 2) this is multi-worker training and auto sharding is
  + enabled.
  Raises:
  ValueError: In case of invalid argument values.
  