src/activations.h
@@ -1,3 +1,4 @@ #include "cuda.h" #ifndef ACTIVATIONS_H #define ACTIVATIONS_H @@ -10,6 +11,12 @@ char *get_activation_string(ACTIVATION a); float activate(float x, ACTIVATION a); 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 void activate_array_ongpu(float *x, int n, ACTIVATION a); void gradient_array_ongpu(float *x, int n, ACTIVATION a, float *delta); #endif #endif