Searched refs:_in_multi_worker_mode (Results 1 – 14 of 14) sorted by relevance
90 if not strategy.extended._in_multi_worker_mode(): # pylint: disable=protected-access115 if (strategy.extended._in_multi_worker_mode() and # pylint: disable=protected-access
1098 if not model._in_multi_worker_mode():
515 if strategy.extended._in_multi_worker_mode(): # pylint: disable=protected-access594 not self._strategy.extended._in_multi_worker_mode()): # pylint: disable=protected-access903 self._strategy.extended._in_multi_worker_mode()): # pylint: disable=protected-access1258 self._strategy.extended._in_multi_worker_mode()): # pylint: disable=protected-access1461 self._strategy.extended._in_multi_worker_mode()): # pylint: disable=protected-access2123 …dataset.element_spec) or strategy.extended._in_multi_worker_mode() # pylint: disable=protected-ac…2153 return strategy.extended._in_multi_worker_mode() or len( # pylint: disable=protected-access
428 def _in_multi_worker_mode(self): member in OneDeviceExtended
736 self.assertTrue(strategy.extended._in_multi_worker_mode())900 self.assertFalse(strategy.extended._in_multi_worker_mode())
636 def _in_multi_worker_mode(self): member in ParameterServerStrategyExtended
2723 def _in_multi_worker_mode(self): member in StrategyExtendedV23744 def _in_multi_worker_mode(self): member in _DefaultDistributionExtended
949 def _in_multi_worker_mode(self): member in CollectiveAllReduceExtended
899 def _in_multi_worker_mode(self): member in MirroredExtended
1652 def _in_multi_worker_mode(self): member in TPUExtended
95 if self._in_multi_worker_mode(): # pylint: disable=protected-access1697 if (self._in_multi_worker_mode() or _is_tpu_multi_host(2804 def _in_multi_worker_mode(self): member in Model2805 …return self.distribute_strategy.extended._in_multi_worker_mode() # pylint: disable=protected-acce…2877 ) and strategy.extended._in_multi_worker_mode() # pylint: disable=protected-access
594 if self._in_multi_worker_mode():2800 def _in_multi_worker_mode(self): member in Model2821 … return strategy and strategy.extended._in_multi_worker_mode() # pylint: disable=protected-access
1686 if (model._in_multi_worker_mode() and
6467 if distribution_strategy.extended._in_multi_worker_mode():