message: Review tpu_strategy.py following feedback
diff: @@ -320,9 +320,9 @@ class TPUStrategyV2(distribute_lib.Strategy):
  """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.
  