experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix doc string for experimental_distribute_values_from_function.

Browse Source 
PiperOrigin-RevId: 301157595
Change-Id: I89479434938e0828e9a0db775f4b5448f30dbab2
This commit is contained in:
Anjali Sridhar 2020-03-16 07:59:36 -07:00 committed by TensorFlower Gardener
parent 636b812bb3
commit 32f5b7dd76
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/distribute/distribute_lib.py
View File

@ -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(