L ops and momentum_sgd ops

rocmarchive / realcaffe2

The repo is obsolete. Use at your own risk.

https://github.com/pytorch/pytorch

Apache License 2.0

12 stars 2 forks source link

Closed ashishfarmer closed 6 years ago

ashishfarmer commented 6 years ago

Making this a WIP PR. Will add the hipified code with intrinsics along with this

ashishfarmer commented 6 years ago

Added the momentum_sgd ops. This PR is ready for review