def METHOD_NAME():
    batch_data = [[], []]
    for data in dev_ds:
        batch_data[0].append(data[0])
        batch_data[1].append(data[1])
        if len(batch_data[0]) == batch_size:
            input_ids = Pad(axis=0, pad_val=0)(batch_data[0])
            segment_ids = Pad(axis=0, pad_val=0)(batch_data[1])
            yield [input_ids, segment_ids]
            batch_data = [[], []]