Created by: jiweibo

PR types

Others

PR changes

Others

Describe

该pr为https://github.com/PaddlePaddle/Paddle/pull/23306 的副本。

典型transformer模型中，while内部的beam_serarch op返回的第三个参数 通过layers.assign的方式传给输入int_idx，while的下一次循环中会用到int_idx，原有逻辑会错误的裁掉assign op导致模型出错。