diff options
author | Qi Liu <liuq901@163.com> | 2016-06-21 15:42:22 +0800 |
---|---|---|
committer | Qi Liu <liuq901@163.com> | 2016-06-21 15:42:22 +0800 |
commit | 3856e63dab1b28aaec4133b6b0ec2a44ebf8cf46 (patch) | |
tree | ed20795f95854c72137bd7537a4097582e997fb3 /nerv/lib/matrix/generic/cumatrix.h | |
parent | fa6700c2caf2a857b95eb62a8dddbf2cfbad2145 (diff) |
complete bp of softmax layer
Diffstat (limited to 'nerv/lib/matrix/generic/cumatrix.h')
-rw-r--r-- | nerv/lib/matrix/generic/cumatrix.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/nerv/lib/matrix/generic/cumatrix.h b/nerv/lib/matrix/generic/cumatrix.h index 459513b..9addae3 100644 --- a/nerv/lib/matrix/generic/cumatrix.h +++ b/nerv/lib/matrix/generic/cumatrix.h @@ -34,6 +34,8 @@ void nerv_matrix_(rowmax_idx)(Matrix *a, Matrix **b, Matrix **idx, CuContext *context, Status *status); void nerv_matrix_(add_row)(Matrix *b, const Matrix *a, double beta, CuContext *context, Status *status); +void nerv_matrix_(add_col)(Matrix *b, const Matrix *a, double beta, + CuContext *context, Status *status); void nerv_matrix_(clip)(Matrix *self, double val1, double val2, CuContext *context, Status *status); void nerv_matrix_(fill)(Matrix *self, double val, |