Searched refs:_CrossReplicaThreadMode (Results 1 – 3 of 3) sorted by relevance
47 class _CrossReplicaThreadMode(_ThreadMode): class
186 distribution_strategy_context._CrossReplicaThreadMode(self._strategy))
273 …self._context = distribution_strategy_context._CrossReplicaThreadMode( # pylint: disable=protecte…2017 …distribution_strategy_context._CrossReplicaThreadMode(self._strategy)) # pylint: disable=protecte…