Esempio n. 1
0
    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)
Esempio n. 2
0
 def _call_for_each_replica(self, fn, args, kwargs):
   return mirrored_run.call_for_each_replica(
       self._container_strategy(), fn, args, kwargs)