update shape equations and ast nodes
debug a NullPointerException when we mis match the shape equation
and the shape propagation result is null. Always be careful with
the situation may have or return null.