From 0f953414dfdbd7abb7b867ce0c3f9390551c1083 Mon Sep 17 00:00:00 2001 From: Determinant Date: Mon, 18 May 2015 11:09:58 +0800 Subject: add sigmoid for cumatrix --- matrix/cukernel.h | 4 ++++ 1 file changed, 4 insertions(+) create mode 100644 matrix/cukernel.h (limited to 'matrix/cukernel.h') diff --git a/matrix/cukernel.h b/matrix/cukernel.h new file mode 100644 index 0000000..5b9e3a6 --- /dev/null +++ b/matrix/cukernel.h @@ -0,0 +1,4 @@ +#ifndef NERV_CUKERNEL_H +#define NERV_CUKERNEL_H +void cuda_sigmoid(const Matrix *a, Matrix *b); +#endif -- cgit v1.2.3