aboutsummaryrefslogtreecommitdiff
path: root/nerv/lib/matrix/cukernel.h
diff options
context:
space:
mode:
authorcloudygoose <cloudygooseg@gmail.com>2015-07-08 08:54:01 +0800
committercloudygoose <cloudygooseg@gmail.com>2015-07-08 08:54:01 +0800
commit72acf24e248cca7d69658d02939f99d57d02e9a9 (patch)
tree0c7e41df72957ee40998503160f9576f1da381a2 /nerv/lib/matrix/cukernel.h
parente599ae7524e0e21e2266f2419555865388bded40 (diff)
added matrix:clip, TODO:testing
Diffstat (limited to 'nerv/lib/matrix/cukernel.h')
-rw-r--r--nerv/lib/matrix/cukernel.h1
1 files changed, 1 insertions, 0 deletions
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);