aboutsummaryrefslogtreecommitdiff
path: root/matrix/cukernel.h
blob: ea81e5ad5903bea55415ec0e4fec9af04cb4084e (plain) (blame)
1
2
3
4
5
6
7
#ifdef NERV_GENERIC_CUKERNEL
void cudak_(cuda_sigmoid)(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