Searched refs:_is_tpu_multi_host (Results 1 – 1 of 1) sorted by relevance
1646 if (self._in_multi_worker_mode() or _is_tpu_multi_host(2758 if _is_tpu_multi_host(strategy):2775 def _is_tpu_multi_host(strategy): function