src/blas.h
@@ -78,6 +78,7 @@ void reorg_ongpu(float *x, int w, int h, int c, int batch, int stride, int forward, float *out); 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); #endif #endif
