Tensorflow Input Function