Searched refs:_push_per_thread_mode (Results 1 – 2 of 2) sorted by relevance
182 _push_per_thread_mode(self._context)1393 _push_per_thread_mode(self._thread_context)1446 _push_per_thread_mode( # thread-local, so not needed with multiple threads1702 _push_per_thread_mode = distribution_strategy_context._push_per_thread_mode # pylint: disable=prot… variable
57 def _push_per_thread_mode(context): function