Merge pull request #46747 from 8bitmp3:patch-1

PiperOrigin-RevId: 356526123 Change-Id: Ie4e5815ba6291dc7505bf08629f355085c50f3e4
tensorflow · Feb 9, 2021 · 9829af6 · 9829af6
2 parents bdf4968 + 38784a8
commit 9829af6
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tensorflow/python/distribute/tpu_strategy.py b/tensorflow/python/distribute/tpu_strategy.py
@@ -322,9 +322,10 @@ def __init__(self,
     """Synchronous training in TPU donuts or Pods.
 
     Args:
-      tpu_cluster_resolver: A tf.distribute.cluster_resolver.TPUClusterResolver,
-        which provides information about the TPU cluster. If None, it will
-        assume running on a local TPU worker.
+      tpu_cluster_resolver: A 
+        `tf.distribute.cluster_resolver.TPUClusterResolver` instance, which
+        provides information about the TPU cluster. If None, it will assume
+        running on a local TPU worker.
       experimental_device_assignment: Optional
         `tf.tpu.experimental.DeviceAssignment` to specify the placement of
         replicas on the TPU cluster.