Created by: JiayiFeng

In WhileGradOpShapeInference, we traverse all its inputs to obtain their dims before further shape inference jobs: https://github.com/PaddlePaddle/Paddle/blob/develop/paddle/operators/while_op.cc#L290

However, some special variables(like lod_rank_table) may not have dims and trying getting this will make a runtime error.