commit | 94d42b03fbffc7352641a50fece7d5d7ea7f69a5 | [log] [tgz] |
---|---|---|
author | Ahmed Taei <ataei@fb.com> | Wed Jun 14 10:59:24 2017 -0700 |
committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Wed Jun 14 11:02:46 2017 -0700 |
tree | 06fa835d1e4cbe5000f2da17820bb4f0522ef680 | |
parent | c1f974aa9fcb44888aa0e0c83dec5d64538ef049 [diff] |
MaxReduction ops GPU implementation. Summary: Move rowwise-max kernel from Softmax to math_util library and implement colwwise-max kernel and MaxReduction ops. Reviewed By: akyrola Differential Revision: D5240329 fbshipit-source-id: a07281a877324de459aace33ff21175a68cfd8f6
Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.
Caffe2 Bay Area Meetup at NVIDIA, May 31 6-8:30, Santa Clara, CA: https://www.meetup.com/Caffe2-Bay-Area/events/239836290/
Caffe2 Community Facebook Group: join to ask questions, talk to other users, and keep informed of important Caffe2 updates.
Please use Github issues (https://github.com/caffe2/caffe2/issues) to ask questions, report bugs, and request new features.
Please participate in our survey (https://www.surveymonkey.com/r/caffe2). We will send you information about new releases and special developer events/webinars.
Caffe2 is released under the BSD 2-Clause license.
Target | Status |
---|---|
Linux | |
Android | |
iOS | |
Linux + MKL |