| commit | 3053618624dcf6ac48d22b0afcb539b2495099c1 | [log] [tgz] |
|---|---|---|
| author | Xiaomeng Yang <bit.yangxm@gmail.com> | Thu Mar 22 00:52:11 2018 -0700 |
| committer | GitHub <noreply@github.com> | Thu Mar 22 00:52:11 2018 -0700 |
| tree | ad89af1605b157c8ba5689dc01d6530e6b1ed7dd | |
| parent | 48c70d2dbdfe6562890c45838437f02eb6bf0976 [diff] |
Add argmax and argmin ops (#2371)
* Revert update on top_k_op
* Add axis to top_k_op
* Remove do { ... } while (false)
* Revert top_k op to upstream
* Add argmin and argmax ops
Add argmin and argmax ops
* Revert top_k_test to upstream
* Add argmin and argmax ops
Add argmin and argmax ops
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 Apache 2.0 license. See the NOTICE file for details.