Searched refs:_get_per_thread_mode (Results 1 – 2 of 2) sorted by relevance
80 def _get_per_thread_mode(): function136 return _get_per_thread_mode().replica_context152 return _get_per_thread_mode().cross_replica_context177 return _get_per_thread_mode().cross_replica_context is not None197 return _get_per_thread_mode().strategy217 per_thread_mode = _get_per_thread_mode()
300 context = _get_per_thread_mode()331 context = _get_per_thread_mode()346 context = _get_per_thread_mode()353 context = _get_per_thread_mode()3610 _get_per_thread_mode = distribution_strategy_context._get_per_thread_mode # pylint: disable=protec… variable