Searched refs:desired_batch_size (Results 1 – 2 of 2) sorted by relevance
392 auto desired_batch_size = dataset()->batch_sizes_[batch_sizes_index_]; in GetNextInternal() local400 while (batch_size < desired_batch_size) { in GetNextInternal()414 int64 slice_end = std::min(offset_ + desired_batch_size - batch_size, in GetNextInternal()438 if ((batch_size == 0 && desired_batch_size > 0) || in GetNextInternal()439 (dataset()->drop_remainder_ && batch_size < desired_batch_size)) { in GetNextInternal()451 if (desired_batch_size == 0) { in GetNextInternal()
188 def _may_form_partial_batches(self, desired_batch_size): argument219 return known_input_batch_dims[0] % desired_batch_size != 0