tree a33914023521df8810ba0107b7640c5541d04702
parent 3b0069a014c05667b6802784c7f1f21936cfcb6c
author Aapo Kyrola <akyrola@fb.com> 1492831084 -0700
committer Yangqing Jia <jiayq@devgpu198.prn2.facebook.com> 1493074345 -0700

CUDA version of PRelu/Gradient + Fix Gradient for dW

Summary:
CUDA version of PRelu and its gradient. Forward pass is straightforward, backward pass requires reductino over the weights.

tsaizhenling, please patch this and test.

Differential Revision: D4931630

fbshipit-source-id: 1238e7d536e41480713865ced91aaef88f4feef5