def get_total_item_size(info, split):
    """Get total item size for given split.
    inputs:
        info = tensorflow dataset info
        split = data split string

    outputs:
        total_item_size = number of total items
    """
    return info.splits[split].num_examples