| commit | 1955d0797e49b88eb9cb003a26191916de993ae4 | [log] [tgz] |
|---|---|---|
| author | Catherine Dong <catherined@fb.com> | Wed Aug 23 16:19:40 2017 -0700 |
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Wed Aug 23 16:33:06 2017 -0700 |
| tree | c5fa03bddcca72957a329f929d098b352c1b7b5c | |
| parent | 2de1bc894b494055e5c6f30d415cd680d993a6df [diff] |
Added fast path for CUDNN global max pooling Summary: This adds a fast path for global max pooling with NCHW. Compared to equivalent ReduceBackMean, this is about 3.5x faster. Based on D5533059. Reviewed By: akyrola Differential Revision: D5681122 fbshipit-source-id: 7a4df934044c7dd01888f095f7dd46654aaf4eae
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 research award competition request for proposals
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.