message: Improve the documentation of InputContext.
diff: @@ -342,12 +342,13 @@ class InputReplicationMode(enum.Enum):
  class InputContext(object):
  """A class wrapping information needed by an input function.
  - This is a context class that is passed to the user's input fn and contains
  - information about the compute replicas and input pipelines. The number of
  - compute replicas (in sync training) helps compute per input pipeline batch
  - size from the desired global batch size. Input pipeline information can be
  - used to return a different subset of the input in each input pipeline (for
  - e.g. shard the input pipeline, use a different input source etc).
  + This is a context class that is passed to the user's input function and
  + contains information about the compute replicas and input pipelines. The
  + number of compute replicas (in sync training) helps compute the local batch
  + size from the desired global batch size for each replica. The input pipeline
  + information can be used to return a different subset of the input in each
  + replica (for e.g. shard the input pipeline, use a different input
  + source etc).
  """
  def __init__(self,
  