Tensorflow Get Model Input Shapes