2014-01-29 04:28:42 +04:00
|
|
|
void pm(int M, int N, float *A);
|
|
|
|
void gemm(int TA, int TB, int M, int N, int K, float ALPHA,
|
|
|
|
float *A, int lda,
|
|
|
|
float *B, int ldb,
|
|
|
|
float BETA,
|
|
|
|
float *C, int ldc);
|
2014-04-11 12:00:27 +04:00
|
|
|
float *random_matrix(int rows, int cols);
|
2014-05-01 03:17:40 +04:00
|
|
|
void time_random_matrix(int TA, int TB, int m, int k, int n);
|
2014-01-29 04:28:42 +04:00
|
|
|
void im2row(float *image, int h, int w, int c, int size, int stride, float *matrix);
|
|
|
|
void im2col(float *image, int h, int w, int c, int size, int stride, float *matrix);
|
|
|
|
void im2col_cpu(float* data_im, const int channels,
|
2014-01-25 02:49:02 +04:00
|
|
|
const int height, const int width, const int ksize, const int stride,
|
2014-01-29 04:28:42 +04:00
|
|
|
float* data_col);
|
|
|
|
void col2im_cpu(float* data_col, const int channels,
|
|
|
|
const int height, const int width, const int ksize, const int stride,
|
|
|
|
float* data_im);
|
|
|
|
void test_blas();
|
2014-04-11 12:00:27 +04:00
|
|
|
|
|
|
|
void gpu_gemm(int TA, int TB, int M, int N, int K, float ALPHA,
|
|
|
|
float *A, int lda,
|
|
|
|
float *B, int ldb,
|
|
|
|
float BETA,
|
|
|
|
float *C, int ldc);
|
|
|
|
void cpu_gemm(int TA, int TB, int M, int N, int K, float ALPHA,
|
|
|
|
float *A, int lda,
|
|
|
|
float *B, int ldb,
|
|
|
|
float BETA,
|
|
|
|
float *C, int ldc);
|
|
|
|
void test_gpu_blas();
|