Skip to content

Commit

Permalink
Merge pull request #30 from SamuelMarks/args-for-google-style-docstrings
Browse files Browse the repository at this point in the history
[*.py] Rename "Arguments:" to "Args:"
  • Loading branch information
jbedorf authored Dec 17, 2020
2 parents e1c7ae0 + 9f061bd commit 671e254
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tensorflow_networking/mpi_collectives/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -136,7 +136,7 @@
def allreduce(tensor, average=True):
"""Perform an MPI allreduce on a tf.Tensor or tf.IndexedSlices.
Arguments:
Args:
tensor: tf.Tensor, tf.Variable, or tf.IndexedSlices to reduce.
The shape of the input must be identical across all ranks.
average: If True, computes the average over all ranks.
Expand Down

0 comments on commit 671e254

Please sign in to comment.