Searched refs:required_tpu (Results 1 – 3 of 3) sorted by relevance
232 "TPU", _get_tpu_strategy_creator(steps_per_run=2), required_tpu=True)236 required_tpu=True)238 "TPUOneStep", _get_tpu_strategy_creator(steps_per_run=1), required_tpu=True)242 required_tpu=True)246 required_tpu=True)250 required_tpu=True,
227 [d.required_tpu or 0 for d in distributions])257 required_tpu=False, argument281 self.required_tpu = required_tpu
86 s for s in all_strategies if s.required_tpu or s.required_gpus or94 s for s in strategies_for_embedding_models() if not s.required_tpu