Searched refs:allow_partial_batch (Results 1 – 3 of 3) sorted by relevance
354 def validate_inputs(x, y, distribution_strategy, allow_partial_batch=False): argument377 if (isinstance(i, dataset_ops.DatasetV2) and not allow_partial_batch):441 allow_partial_batch = (mode == ModeKeys.TRAIN or457 if allow_partial_batch:481 if allow_partial_batch:
176 x, None, model._distribution_strategy, allow_partial_batch=True)186 allow_partial_batch=True)
2164 allow_partial_batch=False): argument2264 drop_remainder = (not allow_partial_batch and