def _call_for_each_replica(self, fn, args, kwargs): # pylint: disable=protected-access return mirrored_strategy._call_for_each_replica( self._container_strategy(), fn, args, kwargs)
def _call_for_each_replica(self, fn, *args, **kwargs): # pylint: disable=protected-access return mirrored_strategy._call_for_each_replica(self, 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(), fn, args, kwargs)
def _call_for_each_replica(self, fn, *args, **kwargs): # pylint: disable=protected-access return mirrored_strategy._call_for_each_replica(self, fn, *args, **kwargs)