Add output defs for fused_adam kernel (#51323)
* add output defs for fused_adam kernel * complete the oters defs for cpu and gpu * remove register for param_out
Showing
想要评论请 注册 或 登录
* add output defs for fused_adam kernel * complete the oters defs for cpu and gpu * remove register for param_out