Stricter rank checking on axes/shape propagation