diff options
| author | Determinant <[email protected]> | 2015-06-10 20:42:10 +0800 |
|---|---|---|
| committer | Determinant <[email protected]> | 2015-06-10 20:42:10 +0800 |
| commit | b818c2562d07a69083377cbc34f2add108e9fa66 (patch) | |
| tree | a595ce4f269035951715334d2942d91d42ae236e /matrix/cukernel.h | |
| parent | c20af45d0756d5d3004105da10e51d42a382ad66 (diff) | |
add CombinerLayer to support branches in NN; add MSELayer
Diffstat (limited to 'matrix/cukernel.h')
| -rw-r--r-- | matrix/cukernel.h | 3 |
1 files changed, 2 insertions, 1 deletions
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 |