Tensorflow Get Model Input Shape In Flowchart