Searched refs:get_real_batch_size (Results 1 – 1 of 1) sorted by relevance
38 def get_real_batch_size(self, dataset_batch): member in PartialBatchPaddingHandler58 original_batch_size = self.get_real_batch_size(dataset_batch)77 self.get_real_batch_size(batch))