diff options
| author | Determinant <ted.sybil@gmail.com> | 2016-05-01 00:44:43 +0800 |
|---|---|---|
| committer | Determinant <ted.sybil@gmail.com> | 2016-05-01 00:44:43 +0800 |
| commit | 34fe366898bc48b4e1ad6fa945dad8821857a459 (patch) | |
| tree | 1210e1a28dd4c0f6e1cefe0b98c95b333ceec158 /nerv/lib/matrix/generic/cumatrix.h | |
| parent | 2bb64ecc5e0350f89cff1e978a1b8dcca1528a06 (diff) | |
| parent | 1aaffa6b9a56d4d580dea7bf3f1b7df1eec5da2f (diff) | |
Merge branch 'master' of ssh://speechlab.sjtu.edu.cn:8022/nerv-dev/nerv
Diffstat (limited to 'nerv/lib/matrix/generic/cumatrix.h')
| -rw-r--r-- | nerv/lib/matrix/generic/cumatrix.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/nerv/lib/matrix/generic/cumatrix.h b/nerv/lib/matrix/generic/cumatrix.h index 79bfc76..459513b 100644 --- a/nerv/lib/matrix/generic/cumatrix.h +++ b/nerv/lib/matrix/generic/cumatrix.h @@ -17,6 +17,11 @@ void nerv_matrix_(tanh)(Matrix *a, const Matrix *b, void nerv_matrix_(tanh_grad)(Matrix *nerr, const Matrix *err, const Matrix *output, CuContext *context, Status *status); +void nerv_matrix_(relu)(Matrix *a, const Matrix *b, + CuContext *context, Status *status); +void nerv_matrix_(relu_grad)(Matrix *nerr, const Matrix *err, + const Matrix *output, + CuContext *context, Status *status); Matrix *nerv_matrix_(softmax)(Matrix *b, const Matrix *a, CuContext *context, Status *status); |