| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | combinations.py | 330 def _get_tpu_strategy_creator(steps_per_run, use_single_core=False, **kwargs): argument
340 strategy = tpu_lib.TPUStrategy(resolver, steps_per_run=steps_per_run,
359 "TPU", _get_tpu_strategy_creator(steps_per_run=2),
362 "TPUOneStep", _get_tpu_strategy_creator(steps_per_run=1),
366 steps_per_run=2, use_single_core=True),
370 steps_per_run=1, use_single_core=True),
|
| D | keras_utils_test.py | 99 steps_per_run = distribution.extended.steps_per_run
100 num_batch_call_per_epoch = steps_per_epoch // steps_per_run
101 if steps_per_epoch % steps_per_run:
518 distribution.extended.steps_per_run > 1):
|
| D | keras_correctness_test_base.py | 293 distribution.extended.steps_per_run > 1):
469 distribution.extended.steps_per_run > 1):
473 update_freq = distribution.extended.steps_per_run
|
| D | metrics_v1_test.py | 107 step_fn, iterator, iterations=distribution.extended.steps_per_run)
114 distribution.extended.steps_per_run)
|
| D | keras_backward_compat_test.py | 1074 distribution.extended.steps_per_run > 1):
|
| D | keras_test.py | 961 distribution.extended.steps_per_run != 1):
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 130 steps_per_run=None, argument
147 self, tpu_cluster_resolver, steps_per_run, device_assignment))
150 def steps_per_run(self): member in TPUStrategy
152 return self._extended.steps_per_run
198 steps_per_run=None, argument
205 if steps_per_run is None:
208 steps_per_run = 1
251 self.steps_per_run = steps_per_run
348 if self.steps_per_run == 1:
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_distributed.py | 314 if (current_strategy.extended.steps_per_run != 1 and
338 current_strategy.extended.steps_per_run)
340 iteration_value = current_strategy.extended.steps_per_run
342 steps_per_run = K.variable(
347 step_fn, iterator, iterations=steps_per_run,
369 steps_to_run = ([current_strategy.extended.steps_per_run] *
371 current_strategy.extended.steps_per_run))
372 if steps_per_epoch % current_strategy.extended.steps_per_run:
374 steps_per_epoch % current_strategy.extended.steps_per_run)
392 steps_per_run.load(step_count, K.get_session())
|
| D | training.py | 147 if (self._distribution_strategy.extended.steps_per_run > 1 and
|
| /external/tensorflow/tensorflow/python/training/ |
| D | basic_session_run_hooks.py | 315 def __init__(self, num_steps=None, last_step=None, steps_per_run=1): argument
343 if steps_per_run is None or steps_per_run < 1:
347 self._steps_per_run_initial_value = steps_per_run
539 def _set_steps_per_run(self, steps_per_run): argument
540 self._steps_per_run = steps_per_run
659 def _set_steps_per_run(self, steps_per_run): argument
660 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 | 786 self.steps_per_run = 5
790 self.train_op = training_util._increment_global_step(self.steps_per_run)
799 save_steps=2*self.steps_per_run,
801 hook._set_steps_per_run(self.steps_per_run)
816 save_steps=2*self.steps_per_run,
818 hook._set_steps_per_run(self.steps_per_run)
858 save_steps=2*self.steps_per_run,
860 hook._set_steps_per_run(self.steps_per_run)
1039 steps_per_run,
1043 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 | 15 name: "steps_per_run"
20 …argspec: "args=[\'self\', \'tpu_cluster_resolver\', \'steps_per_run\', \'device_assignment\'], var…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 15 name: "steps_per_run"
20 …argspec: "args=[\'self\', \'tpu_cluster_resolver\', \'steps_per_run\', \'device_assignment\'], var…
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | async_checkpoint.py | 85 def _set_steps_per_run(self, steps_per_run): argument
86 self._steps_per_run = steps_per_run
|