Fix typos
replace `output_shards_axes` with `output_shard_axes`
diff --git a/tensorflow/python/tpu/tpu.py b/tensorflow/python/tpu/tpu.py
index 83c53aa..9b9d082 100644
--- a/tensorflow/python/tpu/tpu.py
+++ b/tensorflow/python/tpu/tpu.py
@@ -1331,7 +1331,7 @@
... = shard(computation, ...)
If `outputs_from_all_shards` is true, the outputs from all shards of
- `computation` are concatenated back together along their `output_shards_axes`.
+ `computation` are concatenated back together along their `output_shard_axes`.
Otherwise, each output is taken from an arbitrary shard.
Inputs and outputs of the computation must be at least rank-1 Tensors.
@@ -1485,7 +1485,7 @@
as inputs.
If `outputs_from_all_shards` is true, the outputs from all shards of
- `computation` are concatenated back together along their `output_shards_axes`.
+ `computation` are concatenated back together along their `output_shard_axes`.
Otherwise, each output is taken from an arbitrary shard.
Inputs and outputs of the computation must be at least rank-1 Tensors.
