strategize cuDNN activations: ReLU, Sigmoid, TanH
authorEvan Shelhamer <shelhamer@imaginarynumber.net>
Sat, 6 Sep 2014 05:43:58 +0000 (22:43 -0700)
committerEvan Shelhamer <shelhamer@imaginarynumber.net>
Sun, 7 Sep 2014 17:25:23 +0000 (19:25 +0200)
commit14a9198ae1bf75f0de35409a9f14cbb8ece375a9
tree54d1202b2243da54ec2cc653864b73c55634b067
parent00f5fa6bf9287615724a99d7f423807bad576a39
strategize cuDNN activations: ReLU, Sigmoid, TanH
include/caffe/neuron_layers.hpp
src/caffe/layer_factory.cpp
src/caffe/layers/cudnn_relu_layer.cpp [new file with mode: 0644]
src/caffe/layers/cudnn_relu_layer.cu [new file with mode: 0644]
src/caffe/layers/cudnn_sigmoid_layer.cpp [new file with mode: 0644]
src/caffe/layers/cudnn_sigmoid_layer.cu [new file with mode: 0644]
src/caffe/layers/cudnn_tanh_layer.cpp [new file with mode: 0644]
src/caffe/layers/cudnn_tanh_layer.cu [new file with mode: 0644]
src/caffe/test/test_neuron_layer.cpp