Micaela Verucchi
29 lines
602 B
Plaintext
29 lines
602 B
Plaintext
#include "kernels.h"
|
|
|
|
__global__
|
|
void activation_leaky(dnnType *input, dnnType *output, int size, float slope) {
|
|
|
|
int i = blockDim.x*blockIdx.x + threadIdx.x;
|
|
|
|
if(i<size) {
|
|
if (input[i]>0)
|
|
output[i] = input[i];
|
|
else
|
|
output[i] = slope*input[i];
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
ELU activation function
|
|
*/
|
|
void activationLEAKYForward(dnnType* srcData, dnnType* dstData, int size, float slope, cudaStream_t stream)
|
|
{
|
|
int blocks = (size+255)/256;
|
|
int threads = 256;
|
|
|
|
activation_leaky<<<blocks, threads, 0, stream>>>(srcData, dstData, size, slope);
|
|
}
|
|
|
|
|