| | |
| | | #include "opencl.h" |
| | | #include "cuda.h" |
| | | #ifndef ACTIVATIONS_H |
| | | #define ACTIVATIONS_H |
| | | |
| | |
| | | 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); |
| | | void activate_array_ongpu(float *x, int n, ACTIVATION a); |
| | | void gradient_array_ongpu(float *x, int n, ACTIVATION a, float *delta); |
| | | #endif |
| | | |
| | | #endif |