* add output defs for fused_adam kernel

* complete the oters defs for cpu and gpu

* remove register for param_out