diff options
| author | Determinant <[email protected]> | 2015-05-18 23:34:08 +0800 |
|---|---|---|
| committer | Determinant <[email protected]> | 2015-05-18 23:34:08 +0800 |
| commit | 186cf4f39e1c753a6056101f654d2939f812d285 (patch) | |
| tree | da6bbc97c8f553ad473c820484b879a4b6a00968 /matrix/cukernel.h | |
| parent | 23fd2694723ab3f2203e6cd040c5e6633cb989c7 (diff) | |
add softmax for cumatrix
Diffstat (limited to 'matrix/cukernel.h')
| -rw-r--r-- | matrix/cukernel.h | 5 |
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 |