Tensorflow Input Function Definition