Make While Op could run on GPU place and add while_loop unittest (#21672)
1. Make while_op accept GPU conditional data 2. Add more complex test cases for while_loop API
Showing
想要评论请 注册 或 登录
Fork自 PaddlePaddle / Paddle
1. Make while_op accept GPU conditional data 2. Add more complex test cases for while_loop API