diff options
author | Rohan Jain <rohanj@google.com> | 2018-09-28 13:50:12 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-28 14:32:29 -0700 |
commit | 1724d155f00b49bc817189247cbfb0df2092a9da (patch) | |
tree | 3f2606f84d779d8fca28a3d253c70176c4ed3fc1 /tensorflow/contrib/distribute/python/mirrored_strategy.py | |
parent | 64be2ecc07c698df05d88051ec42a0409d1a9863 (diff) |
Automated rollback of commit 7f1d70d97f543d69a9f02cd6df0964f22f9278f3
PiperOrigin-RevId: 214989908
Diffstat (limited to 'tensorflow/contrib/distribute/python/mirrored_strategy.py')
-rw-r--r-- | tensorflow/contrib/distribute/python/mirrored_strategy.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/contrib/distribute/python/mirrored_strategy.py b/tensorflow/contrib/distribute/python/mirrored_strategy.py index 93d42e09a2..4d7516063c 100644 --- a/tensorflow/contrib/distribute/python/mirrored_strategy.py +++ b/tensorflow/contrib/distribute/python/mirrored_strategy.py @@ -484,8 +484,7 @@ class MirroredStrategy(distribute_lib.DistributionStrategy): self._prefetch_on_device, self._auto_shard_dataset) else: return values.PerDeviceDataset( - self._call_dataset_fn(dataset_fn), - self._devices, + self._call_dataset_fn(dataset_fn), self._devices, self._prefetch_on_device) # TODO(priyag): Deal with OutOfRange errors once b/111349762 is fixed. |