Searched refs:_get_per_thread_mode (Results 1 – 2 of 2) sorted by relevance
76 def _get_per_thread_mode(): function189 return _get_per_thread_mode().replica_context205 return _get_per_thread_mode().cross_replica_context230 return _get_per_thread_mode().cross_replica_context is not None250 return _get_per_thread_mode().strategy270 per_thread_mode = _get_per_thread_mode()
301 context = _get_per_thread_mode()332 context = _get_per_thread_mode()347 context = _get_per_thread_mode()354 context = _get_per_thread_mode()3839 _get_per_thread_mode = distribution_strategy_context._get_per_thread_mode # pylint: disable=protec… variable