src/activations.h
@@ -1,3 +1,4 @@ #include "opencl.h" #ifndef ACTIVATIONS_H #define ACTIVATIONS_H @@ -12,6 +13,11 @@ float gradient(float x, ACTIVATION a); void gradient_array(const float *x, const int n, const ACTIVATION a, float *delta); void activate_array(float *x, const int n, const ACTIVATION a); #ifdef GPU cl_kernel get_activation_kernel(); void activate_array_ongpu(cl_mem x, int n, ACTIVATION a); void gradient_array_ongpu(cl_mem x, int n, ACTIVATION a, cl_mem delta); #endif #endif
