message: Only enable last partial support in MirroredStrategy in eager mode.
diff: @@ -381,6 +381,8 @@ class MirroredExtended(distribute_lib.StrategyExtendedV1):
  self._cross_device_ops = cross_device_ops
  self._initialize_strategy(devices)
  + # 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):
  