Home
last modified time | relevance | path

Searched refs:process_batch_and_step_size (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_distributed.py603 batch_size, steps_per_epoch = dist_utils.process_batch_and_step_size(
635 _, validation_steps = dist_utils.process_batch_and_step_size(
699 batch_size, steps = dist_utils.process_batch_and_step_size(
738 batch_size, steps = dist_utils.process_batch_and_step_size(
Dtraining_v2.py208 batch_size, steps_per_epoch = dist_utils.process_batch_and_step_size(
407 batch_size, steps = dist_utils.process_batch_and_step_size(
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistributed_training_utils.py453 def process_batch_and_step_size(strategy, function