search

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

Unified memory for fwd/bwd pass #97

Closed petrex closed 6 years ago

petrex commented 6 years ago

-move pooWs_ to conv_pool base class