src/blas.h
@@ -77,7 +77,8 @@ 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 groups, float temp, float *output, cudaStream_t stream); 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
