prepare_gradient_aggregation for non-leaf output of PartialProgramLayer (#44893)
* 1. add prepare_gradient_aggregation in PartialProgramLayer * 1. draft * fix ci problem
Showing
想要评论请 注册 或 登录
Fork自 PaddlePaddle / Paddle
* 1. add prepare_gradient_aggregation in PartialProgramLayer * 1. draft * fix ci problem