Tensorflow Get Model Inputs