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

Only enable last partial support in MirroredStrategy in eager mode.

Browse Source 
PiperOrigin-RevId: 257686233
This commit is contained in:
Ruoxin Sang 2019-07-11 14:47:16 -07:00 committed by TensorFlower Gardener
parent 14bfae8a04
commit 2cb25a86b2
1 changed files with 3 additions and 1 deletions

4
tensorflow/python/distribute/mirrored_strategy.py
View File

@ -381,7 +381,9 @@ class MirroredExtended(distribute_lib.StrategyExtendedV1):
self._cross_device_ops = cross_device_ops
self._initialize_strategy(devices)
self.experimental_enable_get_next_as_optional = True
# TODO(b/128995245): Enable last partial batch support in graph mode.
if ops.executing_eagerly_outside_functions():
self.experimental_enable_get_next_as_optional = True
def _initialize_strategy(self, devices):
# The _initialize_strategy method is intended to be used by distribute