From 32f5b7dd76eb11608534e5b31f0ab5b3b635fb3d Mon Sep 17 00:00:00 2001
From: Anjali Sridhar <anjalisridhar@google.com>
Date: Mon, 16 Mar 2020 07:59:36 -0700
Subject: [PATCH] Fix doc string for
 `experimental_distribute_values_from_function`.

PiperOrigin-RevId: 301157595
Change-Id: I89479434938e0828e9a0db775f4b5448f30dbab2
---
 tensorflow/python/distribute/distribute_lib.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/python/distribute/distribute_lib.py b/tensorflow/python/distribute/distribute_lib.py
index 6c58633deec..9b819987899 100644
--- a/tensorflow/python/distribute/distribute_lib.py
+++ b/tensorflow/python/distribute/distribute_lib.py
@@ -1377,7 +1377,7 @@ class Strategy(StrategyBase):
         multiple_values.append(tf.constant(1.0))
 
     def value_fn(ctx):
-      return multiple_values[ctx.replica_id]
+      return multiple_values[ctx.replica_id_in_sync_group]
 
     distributed_values = strategy.
       experimental_distribute_values_from_function(