Add argmax_param axis
authorTim Meinhardt <meinhardt.tim@gmail.com>
Tue, 15 Sep 2015 14:55:26 +0000 (16:55 +0200)
committerTim Meinhardt <meinhardt.tim@gmail.com>
Fri, 25 Sep 2015 10:05:44 +0000 (12:05 +0200)
commit6c02c8b7daf123f64b944ede407d0022e98d6e0b
tree1663f70e57a3eb7140d1289baaabb5c8928c47d5
parent674b349522249094b5449f4bc1f90635dc625f4f
Add argmax_param axis
include/caffe/common_layers.hpp
src/caffe/layers/argmax_layer.cpp
src/caffe/proto/caffe.proto