def _call_for_each_replica(self, fn, args, kwargs): self._assert_being_scheduled_by_cluster_coordinator() return mirrored_run.call_for_each_replica(self._container_strategy(), fn, args, kwargs)
def _call_for_each_replica(self, fn, args, kwargs): return mirrored_run.call_for_each_replica( self._container_strategy(), fn, args, kwargs)