| commit | bc3ec13dae840249448bf57f475326a64af93bab | [log] [tgz] |
|---|---|---|
| author | Lukasz Wesolowski <lwesolowski@fb.com> | Thu Apr 27 14:50:24 2017 -0700 |
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Thu Apr 27 15:07:31 2017 -0700 |
| tree | 70fc2f22d51fd38bfd478400a84d027ab61932ea | |
| parent | 12a024241af3f404aabd9fcf3214618e18a90615 [diff] |
change topk operator to use a priority queue Summary: Use a priority queue instead of std::partial_sort to identify the top k elements. This reduces memory usage and improves performance. Differential Revision: D4963931 fbshipit-source-id: 02e75b17ffaf24a4f63c7136626bf0991ee47495
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.
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.
Detailed build matrix (hit refresh if you see icons not showing up due to heroku):
| Target | Status |
|---|---|
| Linux | |
| Mac (CPU) | |
| Android | |
| iOS | |
| Linux + MKL | |
| Windows |