Searched refs:_is_tpu_strategy (Results 1 – 3 of 3) sorted by relevance
265 elif _is_tpu_strategy(strategy) and pure_eager:343 if not _is_tpu_strategy(strategy):367 if _is_tpu_strategy(strategy):395 if _is_tpu_strategy(strategy):422 if _is_tpu_strategy(strategy):485 if _is_tpu_strategy(strategy):545 if _is_tpu_strategy(strategy):674 def _is_tpu_strategy(strategy): function
182 def _is_tpu_strategy(strategy): function
241 def _is_tpu_strategy(strategy): function280 _is_tpu_strategy(distribution_strategy_context.get_strategy())):