Fix gather & stack op (#14355)
* Add int type support for stack_op * Improve gather op to support index with shape N x 1 test=develop * Fix stack_op kernel's registry test=develop
Showing
想要评论请 注册 或 登录
Fork自 PaddlePaddle / Paddle
* Add int type support for stack_op * Improve gather op to support index with shape N x 1 test=develop * Fix stack_op kernel's registry test=develop