def _input_fn():
    dataset = tf.data.TFRecordDataset(tf_records_list, compression_type='GZIP')
    dataset = dataset.map(_parse_proto)
    dataset = dataset.map(_split_xy)
    dataset = dataset.shuffle(buffer_size=(2 * batch_size))
    dataset = dataset.repeat(epochs)
    dataset = dataset.batch(batch_size)
    return dataset
def _input_fn():
    dataset = tf.data.TFRecordDataset(tf_records_list, compression_type='GZIP')
    dataset = dataset.map(_parse_proto)
    dataset = dataset.map(_split_xy)
    dataset = dataset.shuffle(buffer_size=(2 * batch_size))
    dataset = dataset.repeat(epochs)
    dataset = dataset.batch(batch_size)
    return dataset