Searched refs:_get_tpu_strategy_creator (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_combinations.py | 72 def _get_tpu_strategy_creator(steps_per_run, function
232 "TPU", _get_tpu_strategy_creator(steps_per_run=2), required_tpu=True)
235 _get_tpu_strategy_creator(steps_per_run=2, enable_packed_variable=True),
238 "TPUOneStep", _get_tpu_strategy_creator(steps_per_run=1), required_tpu=True)
241 _get_tpu_strategy_creator(steps_per_run=2, use_single_core=True),
245 _get_tpu_strategy_creator(steps_per_run=1, use_single_core=True),
249 _get_tpu_strategy_creator(steps_per_run=2),
|