| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_combinations.py | 76 def _get_tpu_strategy_creator(steps_per_run, argument
123 strategy = tpu_lib.TPUStrategyV1(resolver, steps_per_run,
356 "TPU", _get_tpu_strategy_creator(steps_per_run=2), required_tpu=True)
359 _get_tpu_strategy_creator(steps_per_run=2, enable_packed_variable=True),
364 steps_per_run=2, enable_spmd_xla_paritioning=True),
367 "TPUOneStep", _get_tpu_strategy_creator(steps_per_run=1), required_tpu=True)
370 _get_tpu_strategy_creator(steps_per_run=2, use_single_core=True),
374 _get_tpu_strategy_creator(steps_per_run=1, use_single_core=True),
378 _get_tpu_strategy_creator(steps_per_run=2),
|
| D | tpu_strategy.py | 742 steps_per_run=None, argument
759 self, tpu_cluster_resolver, steps_per_run, device_assignment))
771 def steps_per_run(self): member in TPUStrategyV1
773 return self._extended.steps_per_run
845 steps_per_run=None, argument
853 if steps_per_run is None:
856 steps_per_run = 1
910 self.steps_per_run = steps_per_run
1109 if self.steps_per_run == 1:
|
| D | metrics_v1_test.py | 106 step_fn, iterator, iterations=distribution.extended.steps_per_run)
113 distribution.extended.steps_per_run)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_distributed_v1.py | 163 current_strategy.extended.steps_per_run)
164 steps_per_run = backend.variable(
189 step_fn, iterator, iterations=steps_per_run,
210 steps_to_run = ([current_strategy.extended.steps_per_run] *
212 current_strategy.extended.steps_per_run))
213 if steps_per_epoch % current_strategy.extended.steps_per_run:
215 steps_per_epoch % current_strategy.extended.steps_per_run)
234 backend.get_session().run(steps_per_run.assign(step_count))
|
| D | training.py | 2306 if (self._distribution_strategy.extended.steps_per_run > 1 and
|
| D | training_v1.py | 228 if (self._distribution_strategy.extended.steps_per_run > 1 and
|
| /external/tensorflow/tensorflow/python/training/ |
| D | basic_session_run_hooks.py | 331 def __init__(self, num_steps=None, last_step=None, steps_per_run=1): argument
359 if steps_per_run is None or steps_per_run < 1:
363 self._steps_per_run_initial_value = steps_per_run
573 def _set_steps_per_run(self, steps_per_run): argument
574 self._steps_per_run = steps_per_run
694 def _set_steps_per_run(self, steps_per_run): argument
695 self._steps_per_run = steps_per_run
|
| D | evaluation.py | 80 def __init__(self, num_evals, steps_per_run=1): argument
90 self._steps_per_run_initial_value = steps_per_run
|
| D | basic_session_run_hooks_test.py | 823 self.steps_per_run = 5
827 self.train_op = training_util._increment_global_step(self.steps_per_run)
836 save_steps=2*self.steps_per_run,
838 hook._set_steps_per_run(self.steps_per_run)
853 save_steps=2*self.steps_per_run,
855 hook._set_steps_per_run(self.steps_per_run)
895 save_steps=2*self.steps_per_run,
897 hook._set_steps_per_run(self.steps_per_run)
1075 steps_per_run,
1079 self.train_op = training_util._increment_global_step(steps_per_run)
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 20 name: "steps_per_run"
25 …argspec: "args=[\'self\', \'tpu_cluster_resolver\', \'steps_per_run\', \'device_assignment\'], var…
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | async_checkpoint.py | 90 def _set_steps_per_run(self, steps_per_run): argument
91 self._steps_per_run = steps_per_run
|