| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_combinations.py | 72 def _get_tpu_strategy_creator(steps_per_run, argument
114 strategy = tpu_lib.TPUStrategyV1(resolver, steps_per_run,
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),
|
| D | tpu_strategy.py | 686 steps_per_run=None, argument
703 self, tpu_cluster_resolver, steps_per_run, device_assignment))
715 def steps_per_run(self): member in TPUStrategyV1
717 return self._extended.steps_per_run
789 steps_per_run=None, argument
796 if steps_per_run is None:
799 steps_per_run = 1
853 self.steps_per_run = steps_per_run
1051 if self.steps_per_run == 1:
|
| D | metrics_v1_test.py | 111 step_fn, iterator, iterations=distribution.extended.steps_per_run)
118 distribution.extended.steps_per_run)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_distributed_v1.py | 168 current_strategy.extended.steps_per_run)
169 steps_per_run = K.variable(
194 step_fn, iterator, iterations=steps_per_run,
215 steps_to_run = ([current_strategy.extended.steps_per_run] *
217 current_strategy.extended.steps_per_run))
218 if steps_per_epoch % current_strategy.extended.steps_per_run:
220 steps_per_epoch % current_strategy.extended.steps_per_run)
239 K.get_session().run(steps_per_run.assign(step_count))
|
| D | training.py | 2251 if (self._distribution_strategy.extended.steps_per_run > 1 and
|
| D | training_v1.py | 231 if (self._distribution_strategy.extended.steps_per_run > 1 and
|
| /external/tensorflow/tensorflow/python/training/ |
| D | basic_session_run_hooks.py | 328 def __init__(self, num_steps=None, last_step=None, steps_per_run=1): argument
356 if steps_per_run is None or steps_per_run < 1:
360 self._steps_per_run_initial_value = steps_per_run
559 def _set_steps_per_run(self, steps_per_run): argument
560 self._steps_per_run = steps_per_run
680 def _set_steps_per_run(self, steps_per_run): argument
681 self._steps_per_run = steps_per_run
|
| D | evaluation.py | 84 def __init__(self, num_evals, steps_per_run=1): argument
94 self._steps_per_run_initial_value = steps_per_run
|
| D | basic_session_run_hooks_test.py | 827 self.steps_per_run = 5
831 self.train_op = training_util._increment_global_step(self.steps_per_run)
840 save_steps=2*self.steps_per_run,
842 hook._set_steps_per_run(self.steps_per_run)
857 save_steps=2*self.steps_per_run,
859 hook._set_steps_per_run(self.steps_per_run)
899 save_steps=2*self.steps_per_run,
901 hook._set_steps_per_run(self.steps_per_run)
1079 steps_per_run,
1083 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/keras/distribute/ |
| D | keras_utils_test.py | 110 steps_per_run = distribution.extended.steps_per_run
111 num_batch_call_per_epoch = steps_per_epoch // steps_per_run
112 if steps_per_epoch % steps_per_run:
478 distribution.extended.steps_per_run > 1):
|
| D | keras_correctness_test_base.py | 336 distribution.extended.steps_per_run > 1):
548 distribution.extended.steps_per_run > 1):
552 update_freq = distribution.extended.steps_per_run
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | async_checkpoint.py | 94 def _set_steps_per_run(self, steps_per_run): argument
95 self._steps_per_run = steps_per_run
|