예제 #1
0
 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)
예제 #2
0
 def _call_for_each_replica(self, fn, *args, **kwargs):
   # pylint: disable=protected-access
   return mirrored_strategy._call_for_each_replica(self, fn, *args, **kwargs)
예제 #3
0
 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)
예제 #4
0
 def _call_for_each_replica(self, fn, *args, **kwargs):
   # pylint: disable=protected-access
   return mirrored_strategy._call_for_each_replica(self, fn, *args, **kwargs)