mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Trying some stuff w/ dropout
This commit is contained in:
@ -14,7 +14,9 @@ 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
|
||||
|
Reference in New Issue
Block a user