src/activations.h
@@ -1,3 +1,4 @@ #include "cuda.h" #ifndef ACTIVATIONS_H #define ACTIVATIONS_H @@ -12,6 +13,10 @@ 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
