diff options
author | Jan Horst Hünnemeyer <Jan@Huennemeyer.engineer> | 2018-06-29 14:37:33 +0200 |
---|---|---|
committer | Jan Horst Hünnemeyer <Jan@Huennemeyer.engineer> | 2018-06-29 14:37:33 +0200 |
commit | a69b18a2d24247071b6826be4933226f5eff9af9 (patch) | |
tree | 6c55343361d4162321dd3314512df89e2cb828df /tensorflow/contrib/distribute/python/mirrored_strategy.py | |
parent | b7300de4ef75bdd9373bccc4ae5b98135a70287b (diff) |
Utilizing dict comprehension
Diffstat (limited to 'tensorflow/contrib/distribute/python/mirrored_strategy.py')
-rw-r--r-- | tensorflow/contrib/distribute/python/mirrored_strategy.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/distribute/python/mirrored_strategy.py b/tensorflow/contrib/distribute/python/mirrored_strategy.py index d269bed1e5..cdb505d95d 100644 --- a/tensorflow/contrib/distribute/python/mirrored_strategy.py +++ b/tensorflow/contrib/distribute/python/mirrored_strategy.py @@ -87,7 +87,7 @@ class MirroredStrategy(distribute_lib.DistributionStrategy): self._devices = [device_util.resolve(d) for d in devices] self._canonical_device_set = set(self._devices) self._device_index = values.PerDevice( - dict((d, i) for i, d in enumerate(devices))) + {d: i for i, d in enumerate(devices)}) self._cross_tower_ops = cross_tower_ops self._prefetch_on_device = prefetch_on_device # TODO(yuefengz): consider setting the default device. |