From 44752a8175c54effd0c901333a7eb4cc71dd6810 Mon Sep 17 00:00:00 2001 From: Determinant Date: Wed, 27 May 2015 12:02:44 +0800 Subject: add implementation for sigmoid layer (not tested) --- matrix/cukernel.h | 1 + 1 file changed, 1 insertion(+) (limited to 'matrix/cukernel.h') diff --git a/matrix/cukernel.h b/matrix/cukernel.h index 3cad489..b2b6cb2 100644 --- a/matrix/cukernel.h +++ b/matrix/cukernel.h @@ -1,5 +1,6 @@ #ifdef NERV_GENERIC_CUKERNEL void cudak_(cuda_sigmoid)(const Matrix *a, Matrix *b); +void cudak_(cuda_sigmoid_grad)(const Matrix *output, const Matrix *err, Matrix *nerr); void cudak_(cuda_rowsum)(const Matrix *a, Matrix *b); void cudak_(cuda_rowmax)(const Matrix *a, Matrix *b); void cudak_(cuda_colsum)(const Matrix *a, Matrix *b); -- cgit v1.2.3