From c6ecf1e0420737eafeb99b27b1d716b46a6cbb7a Mon Sep 17 00:00:00 2001 From: Jud White <github@judsonwhite.com> Date: Sun, 25 Mar 2018 20:41:48 +0000 Subject: [PATCH] README.md: add notes to How to compile on Windows --- src/blas.h | 1 + 1 files changed, 1 insertions(+), 0 deletions(-) diff --git a/src/blas.h b/src/blas.h index a5b82ec..e1bfbf0 100644 --- a/src/blas.h +++ b/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); -- Gitblit v1.10.0