src/blas.h
@@ -80,6 +80,7 @@ void softmax_gpu(float *input, int n, int offset, int groups, float temp, float *output); void adam_gpu(int n, float *x, float *m, float *v, float B1, float B2, float rate, float eps, int t); void adam_update_gpu(float *w, float *d, float *m, float *v, float B1, float B2, float eps, float decay, float rate, int n, int batch, int t); void flatten_ongpu(float *x, int spatial, int layers, int batch, int forward, float *out);
