Commit Graph

6 Commits (df50c14abdcf2a1cebe186430756cc7d6c8eacd5)

Author SHA1 Message Date
QI JUN 61ec0b9516
Refine device context (#6433)
7 years ago
Kavya Srinet f52cdaa0ce Updated RMSProp to have learning rate as an input and work with GPU
7 years ago
Kavya Srinet fa12e51675 Adding the default attribute test case
7 years ago
Kavya Srinet 94855f4af0 Fixed changes proposed in the review
7 years ago
Kavya Srinet 163d287143 Made learning rate the input
7 years ago
Kavya Srinet 61c03f9d59 Adding the implementation for rmsprop operator
7 years ago