Searched refs:dataset_batch (Results 1 – 1 of 1) sorted by relevance
38 def get_real_batch_size(self, dataset_batch): argument40 if isinstance(dataset_batch, (tuple, list)):41 dataset_batch = dataset_batch[0]43 assert nest.flatten(dataset_batch)53 return K.cast(K.shape(_find_any_tensor(dataset_batch))[0],56 def update_mask(self, padding_mask, dataset_batch): argument58 original_batch_size = self.get_real_batch_size(dataset_batch)