aboutsummaryrefslogtreecommitdiff
path: root/matrix/cukernel.h
diff options
context:
space:
mode:
authorDeterminant <[email protected]>2015-05-22 12:50:13 +0800
committerDeterminant <[email protected]>2015-05-22 12:50:13 +0800
commit6c0671929f24f296c51568ae3d5c93a3e4d5518f (patch)
tree7f28f155498ec5cf461c7a154032204032d556e1 /matrix/cukernel.h
parent0b128e097d425418499ab2257c5448f14fec3215 (diff)
rename colsum to rowsum; add colsum implementation
Diffstat (limited to 'matrix/cukernel.h')
-rw-r--r--matrix/cukernel.h3
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