From b818c2562d07a69083377cbc34f2add108e9fa66 Mon Sep 17 00:00:00 2001 From: Determinant Date: Wed, 10 Jun 2015 20:42:10 +0800 Subject: add CombinerLayer to support branches in NN; add MSELayer --- matrix/cukernel.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'matrix/cukernel.h') diff --git a/matrix/cukernel.h b/matrix/cukernel.h index 23398c8..8a1494f 100644 --- a/matrix/cukernel.h +++ b/matrix/cukernel.h @@ -14,6 +14,7 @@ void cudak_(cuda_add_row)(const Matrix *a, Matrix *b, double beta); void cudak_(cuda_fill)(Matrix *a, double val); void cudak_(cuda_expand_frm)(const Matrix *a, Matrix *b, int context); void cudak_(cuda_rearrange_frm)(const Matrix *a, Matrix *b, int step); -void cudak_(cuda_scale_row)(const Matrix *a, Matrix *b); +void cudak_(cuda_scale_rows_by_row)(const Matrix *a, Matrix *b); +void cudak_(cuda_scale_rows_by_col)(const Matrix *a, Matrix *b); void cudak_(cuda_decompress)(const Matrix *a, Matrix *b); #endif -- cgit v1.2.3