Created by: jiweibo

PR types

Bug fixes

PR changes

Others

Describe

cherry-pick https://github.com/PaddlePaddle/Paddle/pull/25347

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