Searched refs:is_tpu_strategy (Results 1 – 11 of 11) sorted by relevance
156 if (is_tpu_strategy(strategy) and204 if (is_tpu_strategy(distribution_strategy) and440 def is_tpu_strategy(strategy): function513 not is_tpu_strategy(distribution_strategy))517 and is_tpu_strategy(distribution_strategy)))639 if is_tpu_strategy(strategy):685 if (is_tpu_strategy(model._distribution_strategy) and1131 is_tpu = is_tpu_strategy(strategy)
284 elif K.is_tpu_strategy(distribution) and not context.executing_eagerly():304 not K.is_tpu_strategy(distribution)) or307 elif K.is_tpu_strategy(distribution):
64 def is_tpu_strategy(distribution): function70 if is_tpu_strategy(save_distribution) or is_tpu_strategy(
651 if dist_utils.is_tpu_strategy(model._distribution_strategy):708 if dist_utils.is_tpu_strategy(model._distribution_strategy):745 if dist_utils.is_tpu_strategy(model._distribution_strategy):
353 and dist_utils.is_tpu_strategy(strategy)
216 if distributed_training_utils.is_tpu_strategy(self._distribution_strategy):1981 distributed_training_utils.is_tpu_strategy(2036 if distributed_training_utils.is_tpu_strategy(
228 if distributed_training_utils.is_tpu_strategy(self._distribution_strategy):2136 distributed_training_utils.is_tpu_strategy(2191 if distributed_training_utils.is_tpu_strategy(
81 if (tpu.is_tpu_strategy(strategy) and
193 def is_tpu_strategy(strategy): function203 if not is_tpu_strategy(strategy):
5982 if is_tpu_strategy(distribution_strategy):6014 def is_tpu_strategy(strategy): function
265 if distributed_training_utils.is_tpu_strategy(strategy):