diff options
author | Yuefeng Zhou <yuefengz@google.com> | 2018-05-01 12:24:38 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-01 12:27:25 -0700 |
commit | 59677dc14f5ed28e5d858abc318b1a492f37425f (patch) | |
tree | cec3764f8748b500c8bcce20b3846aff77ebaafc /tensorflow/contrib/distribute/python/mirrored_strategy.py | |
parent | 5c18dc63d752af4a810ed70c6aa18d4f7dd2601a (diff) |
Add device_util.resolve method which merges with current device as well.
PiperOrigin-RevId: 194976633
Diffstat (limited to 'tensorflow/contrib/distribute/python/mirrored_strategy.py')
-rw-r--r-- | tensorflow/contrib/distribute/python/mirrored_strategy.py | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/tensorflow/contrib/distribute/python/mirrored_strategy.py b/tensorflow/contrib/distribute/python/mirrored_strategy.py index 6efd578a77..2e57b02583 100644 --- a/tensorflow/contrib/distribute/python/mirrored_strategy.py +++ b/tensorflow/contrib/distribute/python/mirrored_strategy.py @@ -321,7 +321,6 @@ class MirroredStrategy(distribute_lib.DistributionStrategy): def _fetch(self, val, destination, fn): """Return a copy of `val` or `fn(val)` on `destination`.""" - assert isinstance(destination, six.string_types) if isinstance(val, values.TowerLocalVariable): val = self.reduce(val.reduce_method, val, destinations=destination) with ops.device(destination): |