Commit Graph

8 Commits (29cd089ab20472b66781c2a205eb928137860821)

Author SHA1 Message Date
dangqingqing d432b10d8a Update cuda kernel and doc.
7 years ago
dangqingqing 5bd1e73f5e Refine and speedup momentum operator.
7 years ago
Siddharth Goyal 2e355f032e
Fix attribute naming for momentum_op (#5453)
8 years ago
kexinzhao cb0118f3e5 Polish Operator Doc (m) (#5375)
8 years ago
kavyasrinet 5380a5471b Adding Nesterov Momentum (#4948)
8 years ago
QI JUN 521514da36 fix conflict (#4883)
8 years ago
sidgoyal78 c10da26cf5 Modify implementation
8 years ago
sidgoyal78 d28b3094dd Add momentum operator
8 years ago