yolo layers

src/kernels/activation_leaky.cu

@@ -0,0 +1,29 @@
+#include "kernels.h"
+
+__global__
+void activation_leaky(value_type *input, value_type *output, int size) {
+
+    int i = blockDim.x*blockIdx.x + threadIdx.x;
+
+    if(i<size) {    
+        if (input[i]>0)
+            output[i] = input[i];
+        else
+            output[i] = 0.1f*input[i];
+    }
+ }
+
+
+/**
+    ELU activation function
+*/
+void activationLEAKYForward(value_type* srcData, value_type* dstData, int size)
+{
+    int blocks = (size+255)/256;
+    int threads = 256;
+    
+    activation_leaky<<<blocks, threads>>>(srcData, dstData, size);
+    checkCuda( cudaDeviceSynchronize() );
+}
+
+