Searched refs:_push_per_thread_mode (Results 1 – 2 of 2) sorted by relevance
61 def _push_per_thread_mode(context): function
393 _push_per_thread_mode(self._context)2934 _push_per_thread_mode(self._thread_context)2989 _push_per_thread_mode( # thread-local, so not needed with multiple threads3609 _push_per_thread_mode = distribution_strategy_context._push_per_thread_mode # pylint: disable=prot… variable