src/activations.h @@ -1,6 +1,7 @@ #include "cuda.h" #ifndef ACTIVATIONS_H #define ACTIVATIONS_H #include "cuda.h" #include "math.h" typedef enum{ LOGISTIC, RELU, RELIE, LINEAR, RAMP, TANH, PLSE