def fetch_dataset(filename):
    buffer_size = ((8 * 1024) * 1024)
    dataset = tf.data.TFRecordDataset(filename, buffer_size=buffer_size)
    return dataset
def fetch_dataset(filename):
    buffer_size = ((8 * 1024) * 1024)
    dataset = tf.data.TFRecordDataset(filename, buffer_size=buffer_size)
    return dataset