def _call_for_each_replica(self, fn, args, kwargs):
     # pylint: disable=protected-access
     return mirrored_strategy._call_for_each_replica(
         self._container_strategy(), self._device_map, fn, args, kwargs)
 def _call_for_each_replica(self, fn, args, kwargs):
   # pylint: disable=protected-access
   return mirrored_strategy._call_for_each_replica(
       self._container_strategy(), self._device_map, fn, args, kwargs)