Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/engine/
Ddistributed_training_utils.py354 def validate_inputs(x, y, distribution_strategy, allow_partial_batch=False): argument
377 if (isinstance(i, dataset_ops.DatasetV2) and not allow_partial_batch):
441 allow_partial_batch = (mode == ModeKeys.TRAIN or
457 if allow_partial_batch:
481 if allow_partial_batch:
Dtraining_distributed.py176 x, None, model._distribution_strategy, allow_partial_batch=True)
186 allow_partial_batch=True)
Dtraining.py2164 allow_partial_batch=False): argument
2264 drop_remainder = (not allow_partial_batch and