aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/distribute/python/mirrored_strategy.py
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-06-21 14:02:20 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-06-21 14:05:28 -0700
commit8693cc2d29e3f62a15e33801c93af73e752ba14c (patch)
treec99b490bab80f56acda2130e390f798a4a5253fa /tensorflow/contrib/distribute/python/mirrored_strategy.py
parent92e135e07949d6d1291b87a9116e345f1163d4d1 (diff)
Delete DistributionStrategy.fetch() now that it is unused.
PiperOrigin-RevId: 201582230
Diffstat (limited to 'tensorflow/contrib/distribute/python/mirrored_strategy.py')
-rw-r--r--tensorflow/contrib/distribute/python/mirrored_strategy.py27
1 files changed, 0 insertions, 27 deletions
diff --git a/tensorflow/contrib/distribute/python/mirrored_strategy.py b/tensorflow/contrib/distribute/python/mirrored_strategy.py
index d8668b398f..98fea76b3d 100644
--- a/tensorflow/contrib/distribute/python/mirrored_strategy.py
+++ b/tensorflow/contrib/distribute/python/mirrored_strategy.py
@@ -351,33 +351,6 @@ class MirroredStrategy(distribute_lib.DistributionStrategy):
assert isinstance(tower_local_var, values.Mirrored)
return array_ops.identity(tower_local_var.get())
- def _fetch(self, val, destination, fn):
- """Return a copy of `val` or `fn(val)` on `destination`."""
- if isinstance(val, values.TowerLocalVariable):
- val = self.reduce(val.reduce_method, val, destinations=destination)
- with ops.device(destination):
- return fn(self.unwrap(val)[0])
-
- assert isinstance(val, values.Mirrored), (
- "val = %s (type %s)" % (val, val.__class__.__name__))
- if val.on_device(destination):
- with ops.device(destination):
- # Use an identity here to make sure we are returning a tensor
- # instead of e.g. a variable object.
- return array_ops.identity(fn(val.get(destination)))
- device = None
- for d in self._devices:
- if val.on_device(d):
- device = d
- break
- assert device is not None, (
- "Could not find destination %s in list of devices %s." %
- (destination, val.devices))
- with ops.device(device):
- v = fn(val.get(device))
- with ops.device(destination):
- return array_ops.identity(v)
-
def _unwrap(self, val):
if isinstance(val, values.DistributedValues):
# Return in a deterministic order.