darknet/src/mini_blas.h

33 lines
1.3 KiB
C
Raw Normal View History

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);
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-05-03 02:20:34 +04:00
void im2col_gpu(float* data_im, const int channels,
const int height, const int width, const int ksize, const int stride,
float* data_col);
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,
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-05-03 02:20:34 +04:00
void gemm_gpu(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-05-03 02:20:34 +04:00
void gemm_cpu(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-05-03 02:20:34 +04:00
void axpy_cpu(int N, float ALPHA, float *X, int INCX, float *Y, int INCY);
void scal_cpu(int N, float ALPHA, float *X, int INCX);
void test_gpu_blas();