From 252e3b1916cfaca0783c9e90efaa55eb07b1a8cd Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Sun, 06 Nov 2016 00:27:31 +0000 Subject: [PATCH] :charizard: :charizard: :charizard: --- src/blas.h | 7 +++++++ 1 files changed, 7 insertions(+), 0 deletions(-) diff --git a/src/blas.h b/src/blas.h index 95374c9..51554a8 100644 --- a/src/blas.h +++ b/src/blas.h @@ -34,7 +34,11 @@ void l2_cpu(int n, float *pred, float *truth, float *delta, float *error); void weighted_sum_cpu(float *a, float *b, float *s, int num, float *c); +void softmax(float *input, int n, float temp, float *output); + #ifdef GPU +#include "cuda.h" + void axpy_ongpu(int N, float ALPHA, float * X, int INCX, float * Y, int INCY); void axpy_ongpu_offset(int N, float ALPHA, float * X, int OFFX, int INCX, float * Y, int OFFY, int INCY); void copy_ongpu(int N, float * X, int INCX, float * Y, int INCY); @@ -73,5 +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 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 -- Gitblit v1.10.0