Implement ArgMaxLayer forward_cpu and reshape for axis param
authorTim Meinhardt <meinhardt.tim@gmail.com>
Tue, 15 Sep 2015 14:56:16 +0000 (16:56 +0200)
committerTim Meinhardt <meinhardt.tim@gmail.com>
Fri, 25 Sep 2015 10:05:54 +0000 (12:05 +0200)
commitc77d5e5156f94720c1decd13f7f87fe78df9d4eb
tree336c049da64830427e267248b651da70d57cb387
parent6c02c8b7daf123f64b944ede407d0022e98d6e0b
Implement ArgMaxLayer forward_cpu and reshape for axis param
src/caffe/layers/argmax_layer.cpp