strategize cuDNN softmax
authorEvan Shelhamer <shelhamer@imaginarynumber.net>
Sat, 6 Sep 2014 06:53:04 +0000 (23:53 -0700)
committerEvan Shelhamer <shelhamer@imaginarynumber.net>
Sun, 7 Sep 2014 17:56:15 +0000 (19:56 +0200)
commit84bd1f5e98b9bd59be96f7af8c116123cee967f8
treed7e5872333e798840fa869e8ec2556b8483c729b
parent14a9198ae1bf75f0de35409a9f14cbb8ece375a9
strategize cuDNN softmax
include/caffe/common_layers.hpp
src/caffe/layer_factory.cpp
src/caffe/layers/cudnn_softmax_layer.cpp [new file with mode: 0644]
src/caffe/layers/cudnn_softmax_layer.cu [new file with mode: 0644]
src/caffe/layers/softmax_layer.cpp
src/caffe/test/test_softmax_layer.cpp