aboutsummaryrefslogtreecommitdiff
path: root/nerv/lib/matrix/generic/cumatrix.h
diff options
context:
space:
mode:
authorQi Liu <liuq901@163.com>2016-03-11 21:28:29 +0800
committerQi Liu <liuq901@163.com>2016-03-11 21:28:29 +0800
commit442e261a0f2cb8836e2859bd814a267cc8aa5db2 (patch)
tree112dd3932a8d23fc2e36f67c347f13bb2d19232a /nerv/lib/matrix/generic/cumatrix.h
parente2a9af061db485d4388902d738c9d8be3f94ab34 (diff)
parent14c1997203e04838b1737716dc385e1aa08fe91f (diff)
update diagonlal lstm
Diffstat (limited to 'nerv/lib/matrix/generic/cumatrix.h')
-rw-r--r--nerv/lib/matrix/generic/cumatrix.h2
1 files changed, 2 insertions, 0 deletions
diff --git a/nerv/lib/matrix/generic/cumatrix.h b/nerv/lib/matrix/generic/cumatrix.h
index fe83b5d..de3a09e 100644
--- a/nerv/lib/matrix/generic/cumatrix.h
+++ b/nerv/lib/matrix/generic/cumatrix.h
@@ -33,6 +33,8 @@ void nerv_matrix_(clip)(Matrix *self, double val1, double val2,
CuContext *context, Status *status);
void nerv_matrix_(fill)(Matrix *self, double val,
CuContext *context, Status *status);
+void nerv_matrix_(diagonalize)(Matrix *self,
+ CuContext *context, Status *status);
void nerv_matrix_(copy_fromd)(Matrix *a, const Matrix *b,
int a_begin, int b_begin, int b_end,
CuContext *context, Status *status);