Home
last modified time | relevance | path

Searched refs:is_tpu_strategy (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/keras/engine/
Ddistributed_training_utils.py375 if is_tpu_strategy(distribution_strategy):
394 def is_tpu_strategy(strategy): function
443 and is_tpu_strategy(distribution_strategy)))
Dtraining_distributed.py106 if distributed_training_utils.is_tpu_strategy(model._distribution_strategy):
155 if distributed_training_utils.is_tpu_strategy(model._distribution_strategy):
187 if distributed_training_utils.is_tpu_strategy(model._distribution_strategy):
Dtraining.py146 if distributed_training_utils.is_tpu_strategy(self._distribution_strategy):
1970 is_tpu = distributed_training_utils.is_tpu_strategy(
2200 distributed_training_utils.is_tpu_strategy(
2205 if (self.stateful and distributed_training_utils.is_tpu_strategy(
/external/tensorflow/tensorflow/python/keras/
Dbackend.py5341 if is_tpu_strategy(distribution_strategy):
5373 def is_tpu_strategy(strategy): function
/external/tensorflow/tensorflow/contrib/distribute/python/
Dkeras_test.py960 if (distributed_training_utils.is_tpu_strategy(distribution) and