Searched refs:_get_per_thread_mode (Results 1 – 2 of 2) sorted by relevance
76 def _get_per_thread_mode(): function123 return _get_per_thread_mode().replica_context157 return _get_per_thread_mode().cross_replica_context171 return _get_per_thread_mode().cross_replica_context is not None191 return _get_per_thread_mode().strategy205 per_thread_mode = _get_per_thread_mode()
98 context = _get_per_thread_mode()127 context = _get_per_thread_mode()142 context = _get_per_thread_mode()149 context = _get_per_thread_mode()1703 _get_per_thread_mode = distribution_strategy_context._get_per_thread_mode # pylint: disable=protec… variable