Tensorflow Get Model Input Shape