summaryrefslogtreecommitdiff
path: root/src/caffe/layers/argmax_layer.cpp
AgeCommit message (Collapse)AuthorFilesLines
2015-12-01dismantle layer headersEvan Shelhamer1-1/+1
No more monolithic includes: split layers into their own headers for modular inclusion and build.
2015-11-06Fix ArgMaxLayer::Reshape for any num of bottom axesTim Meinhardt1-1/+3
2015-10-21Clean redundant/unnecessary headersKang Kim1-2/+1
2015-09-25Implement ArgMaxLayerTest for axis paramTim Meinhardt1-13/+15
2015-09-25Implement ArgMaxLayer forward_cpu and reshape for axis paramTim Meinhardt1-15/+38
2015-09-25Add argmax_param axisTim Meinhardt1-5/+17
2015-02-05Layer type is a stringJeff Donahue1-1/+1
2014-10-13move registration code to corresponding cpp files.Yangqing Jia1-0/+1
2014-09-19fix types of (Layer)SetUp, Reshape, Forward, and Backward callsJonathan L Long1-8/+8
Using the type vector<Blob<Dtype*>* for outputs allows modification of the vector itself, while it is only okay to modify the blobs pointed to by the elements of the vector. Switching the types to const vector<Blob<Dtype>*>& makes them more correct.
2014-09-18split off Reshape for common layersJonathan L Long1-0/+5
2014-08-13Generalize loss by allowing any top blob to be used as a loss in whichJeff Donahue1-4/+2
its elements are summed with a scalar coefficient. Forward for layers no longer returns a loss; instead all loss layers must have top blobs. Existing loss layers are given a top blob automatically by Net::Init, with an associated top_loss_weight of 1 (set in LossLayer::FurtherSetUp). Due to the increased amount of common SetUp logic, the SetUp interface is modified such that all subclasses should normally override FurtherSetUp only, which is called by SetUp.
2014-08-06LICENSE governs the whole project so strip file headersEvan Shelhamer1-2/+0
2014-07-20remove unused includes in AccuracyLayer and ArgMaxLayerJonathan L Long1-2/+0
2014-07-20Fix style issues in accuracy & argmax layerKai Li1-15/+8
2014-07-20Limit the comparison functions to have file scopeKai Li1-1/+1
2014-07-20Move compararing function from common_layers to argmax_layerKai Li1-1/+9
2014-07-20Use std::partial_sort in the ArgMaxLayer as suggested by @shuokayKai Li1-2/+3
2014-07-20Simplify the top-k argmax layer using std::sortKai Li1-30/+8
2014-07-20Extend the ArgMaxLayer to output top k resultsKai Li1-13/+41
2014-06-20remove erroneous comment in ArgMaxLayerJonathan L Long1-1/+0
2014-06-08layers declare their names and number of input/output blobs, and don'tJeff Donahue1-2/+1
check top/bottom blob counts explicitly in SetUp; instead call base Layer::SetUp.
2014-05-20Fixed lint errors due to ArgmaxLayerSergey Karayev1-6/+5
2014-05-20Added FLT_MAX to argmax layerSergio Guadarrama1-2/+1
2014-05-20Added ArgMax LayerSergio Guadarrama1-0/+59
Conflicts: src/caffe/proto/caffe.proto