diff options
| author | Determinant <[email protected]> | 2015-05-28 16:14:21 +0800 |
|---|---|---|
| committer | Determinant <[email protected]> | 2015-05-28 16:14:21 +0800 |
| commit | e4dedc2992149d245ea65132131253072d3276b8 (patch) | |
| tree | 31055890c6b662df9091dd7272556a69f216de28 /matrix/cukernel.h | |
| parent | c13115662e739b434f1071eb623a41a39d8b4985 (diff) | |
add mul_elem and log_elem; add dimension checking in cumatrix methods
Diffstat (limited to 'matrix/cukernel.h')
| -rw-r--r-- | matrix/cukernel.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/matrix/cukernel.h b/matrix/cukernel.h index b2b6cb2..232699d 100644 --- a/matrix/cukernel.h +++ b/matrix/cukernel.h @@ -1,4 +1,6 @@ #ifdef NERV_GENERIC_CUKERNEL +void cudak_(cuda_mul_elem)(const Matrix *a, const Matrix *b, Matrix *c); +void cudak_(cuda_log_elem)(const Matrix *a, Matrix *b); void cudak_(cuda_sigmoid)(const Matrix *a, Matrix *b); void cudak_(cuda_sigmoid_grad)(const Matrix *output, const Matrix *err, Matrix *nerr); void cudak_(cuda_rowsum)(const Matrix *a, Matrix *b); |