From 72acf24e248cca7d69658d02939f99d57d02e9a9 Mon Sep 17 00:00:00 2001 From: cloudygoose Date: Wed, 8 Jul 2015 08:54:01 +0800 Subject: added matrix:clip, TODO:testing --- nerv/lib/matrix/cukernel.h | 1 + 1 file changed, 1 insertion(+) (limited to 'nerv/lib/matrix/cukernel.h') diff --git a/nerv/lib/matrix/cukernel.h b/nerv/lib/matrix/cukernel.h index 8a1494f..7bb4c2c 100644 --- a/nerv/lib/matrix/cukernel.h +++ b/nerv/lib/matrix/cukernel.h @@ -12,6 +12,7 @@ void cudak_(cuda_softmax_denominator)(const Matrix *a, const Matrix *max, Matrix void cudak_(cuda_softmax_final)(const Matrix *a, const Matrix *max, const Matrix *deno, Matrix *b); void cudak_(cuda_add_row)(const Matrix *a, Matrix *b, double beta); void cudak_(cuda_fill)(Matrix *a, double val); +void cudak_(cuda_clip)(Matrix *a, double val_1, double val_2); 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_rows_by_row)(const Matrix *a, Matrix *b); -- cgit v1.2.3