aboutsummaryrefslogtreecommitdiff
path: root/matrix/cukernel.h
diff options
context:
space:
mode:
authorDeterminant <[email protected]>2015-05-18 23:34:08 +0800
committerDeterminant <[email protected]>2015-05-18 23:34:08 +0800
commit186cf4f39e1c753a6056101f654d2939f812d285 (patch)
treeda6bbc97c8f553ad473c820484b879a4b6a00968 /matrix/cukernel.h
parent23fd2694723ab3f2203e6cd040c5e6633cb989c7 (diff)
add softmax for cumatrix
Diffstat (limited to 'matrix/cukernel.h')
-rw-r--r--matrix/cukernel.h5
1 files changed, 4 insertions, 1 deletions
diff --git a/matrix/cukernel.h b/matrix/cukernel.h
index f86a69b..9c13558 100644
--- a/matrix/cukernel.h
+++ b/matrix/cukernel.h
@@ -1,5 +1,8 @@
#ifndef NERV_CUKERNEL_H
#define NERV_CUKERNEL_H
void cuda_sigmoid(const Matrix *a, Matrix *b);
-void cuda_rowsum(const Matrix *a, Matrix *b);
+void cuda_colsum(const Matrix *a, Matrix *b);
+void cuda_colmax(const Matrix *a, Matrix *b);
+void cuda_softmax_denominator(const Matrix *a, const Matrix *max, Matrix *b);
+void cuda_softmax_final(const Matrix *a, const Matrix *max, const Matrix *deno, Matrix *b);
#endif