Searched refs:is_tpu_strategy (Results 1 – 12 of 12) sorted by relevance
56 is_tpu = backend.is_tpu_strategy(strategy)
280 elif K.is_tpu_strategy(distribution) and not context.executing_eagerly():296 if ((context.executing_eagerly() and not K.is_tpu_strategy(distribution)) or299 elif K.is_tpu_strategy(distribution):
156 if (K.is_tpu_strategy(strategy) and204 if (K.is_tpu_strategy(distribution_strategy) and499 not K.is_tpu_strategy(distribution_strategy))504 K.is_tpu_strategy(distribution_strategy)))626 if K.is_tpu_strategy(strategy):672 if (K.is_tpu_strategy(model._distribution_strategy) and
68 if backend.is_tpu_strategy(save_distribution) or backend.is_tpu_strategy(
172 if backend.is_tpu_strategy(distribution):513 if backend.is_tpu_strategy(distribution):566 if backend.is_tpu_strategy(distribution):1010 if backend.is_tpu_strategy(distribution):1093 if mode == 'graph' and backend.is_tpu_strategy(distribution):1126 if mode == 'eager' and backend.is_tpu_strategy(distribution):1129 if mode == 'graph' and backend.is_tpu_strategy(distribution):
653 if K.is_tpu_strategy(model._distribution_strategy):710 if K.is_tpu_strategy(model._distribution_strategy):747 if K.is_tpu_strategy(model._distribution_strategy):
230 if K.is_tpu_strategy(self._distribution_strategy):2139 K.is_tpu_strategy(self._distribution_strategy)):2193 if K.is_tpu_strategy(strategy) and not drop_remainder:
2250 if backend.is_tpu_strategy(self._distribution_strategy):2776 return (backend.is_tpu_strategy(strategy) and
80 if (backend.is_tpu_strategy(strategy) and
214 def is_tpu_strategy(strategy: Any) -> bool: function225 if not is_tpu_strategy(strategy):
6517 if is_tpu_strategy(distribution_strategy):6549 def is_tpu_strategy(strategy): function
302 if K.is_tpu_strategy(strategy):