Moved the loop inside PReLUParamBackward to do the reduction inside the kernel
authorMausoom Sarkar <msarkar@adobe.com>
Tue, 13 Oct 2015 13:05:32 +0000 (18:35 +0530)
committerMausoom Sarkar <msarkar@adobe.com>
Mon, 19 Oct 2015 08:21:40 +0000 (13:51 +0530)
commit4c93b3dc555891ae0ad75092b6c0f77508740ecf
treec24e85107d4ff5b62fdef45a4471580a82a9669c
parent0151742509c23b927b0159230ac70399ac4f8967
Moved the loop inside PReLUParamBackward to do the reduction inside the kernel
Now PReLU backward is taking the same time as forward

Code cleanup
Removed unnecessary code
Fixed indent
merge if(channed_shared_)
src/caffe/layers/prelu_layer.cu