Searched refs:pad_batch (Results 1 – 2 of 2) sorted by relevance
64 def pad_batch(self, *dataset_batch_elements): member in PartialBatchPaddingHandler
611 dataset = dataset.map(padding_handler.pad_batch)