aboutsummaryrefslogblamecommitdiff
path: root/nerv/lib/matrix/generic/cumatrix.h
blob: 5b8076f362f0bae71602095b00a5bc4e2dc53e31 (plain) (tree) 
8b2aa75

5e407d7










ca3500f



5e407d7











72acf24

5e407d7



















1a9f63e


c1f31af


9a17267





5e407d7









0ee43c2

a7608a6




9e7171e







1

2
3
4
5
6
7
8
9
10
11

12
13
14

15
16
17
18
19
20
21
22
23
24
25

26

27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

46
47

48
49

50
51
52
53
54

55
56
57
58
59
60
61
62
63

64

65
66
67
68

69
70
71
72
73
74
75

                         









                                                                       


                                                                      










                                                                          
                                                                                  


















                                                                                

                                                                            

                                                                            




                                                                                                                                          








                                                                
                                                                             



                                                                      






                                                                                              
#include "../../common.h"

void nerv_matrix_(add)(Matrix *c, const Matrix *a, const Matrix *b,
                            MATRIX_ELEM alpha, MATRIX_ELEM beta,
                            Status *status);
void nerv_matrix_(mul)(Matrix *c, const Matrix *a, const Matrix *b,
                            MATRIX_ELEM alpha, MATRIX_ELEM beta,
                            int ta, int tb, Status *status);
void nerv_matrix_(sigmoid)(Matrix *a, const Matrix *b, Status *status);
void nerv_matrix_(sigmoid_grad)(Matrix *nerr, const Matrix *err,
                                const Matrix *output, Status *status);
void nerv_matrix_(tanh)(Matrix *a, const Matrix *b, Status *status);
void nerv_matrix_(tanh_grad)(Matrix *nerr, const Matrix *err,
                                const Matrix *output, Status *status);

Matrix *nerv_matrix_(softmax)(Matrix *b, const Matrix *a, Status *status);
Matrix *nerv_matrix_(rowsum)(Matrix *a, Status *status);
Matrix *nerv_matrix_(colsum)(Matrix *a, Status *status);
Matrix *nerv_matrix_(colsame)(Matrix *a, const Matrix *ref,
                                Status *status);
Matrix *nerv_matrix_(rowmax)(Matrix *a, Status *status);
void nerv_matrix_(rowmax_idx)(Matrix *a, Matrix **b, Matrix **idx,
                                Status *status);
void nerv_matrix_(add_row)(Matrix *b, const Matrix *a, double beta,
                            Status *status);
void nerv_matrix_(clip)(Matrix *self, double val_1, double val_2, Status *status);
void nerv_matrix_(fill)(Matrix *self, double val, Status *status);
void nerv_matrix_(copy_fromd)(Matrix *a, const Matrix *b,
                            int a_begin, int b_begin, int b_end,
                            Status *status);
void nerv_matrix_(copy_fromh)(Matrix *a, const Matrix *b,
                            int a_begin, int b_begin, int b_end,
                            Status *status);
void nerv_matrix_(copy_toh)(Matrix *a, const Matrix *b,
                            int a_begin, int a_end, int b_begin,
                            Status *status);
Matrix *nerv_matrix_(trans)(Matrix *a, Status *status);
void nerv_matrix_(mul_elem)(Matrix *c, const Matrix *a, const Matrix *b,
                            Status *status);

void nerv_matrix_(log_elem)(Matrix *b, const Matrix *a, Status *status);

Matrix *nerv_matrix_(decompress)(const Matrix *a, int orig_col, Status *status);
void nerv_matrix_(copy_rows_fromh_by_idx)(Matrix *a, const Matrix *b,
                            const Matrix *idx, int b_begin, Status *status);
void nerv_matrix_(copy_rows_fromd_by_idx)(Matrix *a, const Matrix *b,
                            const Matrix *idx, int b_begin, Status *status);
void nerv_matrix_(copy_rows_fromd_by_colidx)(Matrix *a, const Matrix *b,
                            const Matrix *idx, int b_begin, Status *status);

#ifdef __NERV_FUTURE_CUDA_7
void nerv_matrix_(update_select_rows_by_rowidx)(Matrix *c, const Matrix *a, const Matrix *idx, double alpha, double beta, Status *status);
void nerv_matrix_(update_select_rows_by_colidx)(Matrix *c, const Matrix *a, const Matrix *idx, double alpha, double beta, Status *status);
#endif

void nerv_matrix_(expand_frm)(Matrix *a, const Matrix *b,
                            int context, Status *status);
void nerv_matrix_(rearrange_frm)(Matrix *a, const Matrix *b,
                                int step, Status *status);
void nerv_matrix_(scale_rows_by_col)(Matrix *a, const Matrix *b,
                                    Status *status);
void nerv_matrix_(scale_rows_by_row)(Matrix *a, const Matrix *b,
                                    Status *status);
void nerv_matrix_(prefixsum_row)(Matrix *a, const Matrix *b, Status *status);
void nerv_matrix_(thres_mask)(Matrix *a, Matrix *b,
                                double thres, double low, double high,
                                Status *status);
void nerv_matrix_(rand_uniform)(Matrix *a, Status *status);

#ifdef __NERV_FUTURE_CUDA_7
void nerv_matrix_(update_select_rows_by_rowidx)(Matrix *c, const Matrix *a, const Matrix *idx,
                                                double alpha, double beta, Status *status);
void nerv_matrix_(update_select_rows_by_colidx)(Matrix *c, const Matrix *a, const Matrix *idx,
                                                double alpha, double beta, Status *status);
#endif
Powered by cgit, Themed by Ted Yin.