Searched refs:_may_form_partial_batches (Results 1 – 1 of 1) sorted by relevance
183 if self._may_form_partial_batches(new_batch_dim):188 def _may_form_partial_batches(self, desired_batch_size): member in _RebatchDataset