How to split dataset and feed into input_fn

Question

I have successfully read two csv files using make_csv_dataset within a input_fn() and passed that into a tf.estimator.

I did it by first splitti