diff options
| author | Determinant <ted.sybil@gmail.com> | 2015-05-22 12:50:13 +0800 |
|---|---|---|
| committer | Determinant <ted.sybil@gmail.com> | 2015-05-22 12:50:13 +0800 |
| commit | 6c0671929f24f296c51568ae3d5c93a3e4d5518f (patch) | |
| tree | 7f28f155498ec5cf461c7a154032204032d556e1 /matrix/cukernel.h | |
| parent | 0b128e097d425418499ab2257c5448f14fec3215 (diff) | |
rename colsum to rowsum; add colsum implementation
Diffstat (limited to 'matrix/cukernel.h')
| -rw-r--r-- | matrix/cukernel.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/matrix/cukernel.h b/matrix/cukernel.h index ea81e5a..67a255e 100644 --- a/matrix/cukernel.h +++ b/matrix/cukernel.h @@ -1,7 +1,8 @@ #ifdef NERV_GENERIC_CUKERNEL void cudak_(cuda_sigmoid)(const Matrix *a, Matrix *b); +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); -void cudak_(cuda_colmax)(const Matrix *a, Matrix *b); void cudak_(cuda_softmax_denominator)(const Matrix *a, const Matrix *max, Matrix *b); void cudak_(cuda_softmax_final)(const Matrix *a, const Matrix *max, const Matrix *deno, Matrix *b); #endif |