mirror of https://github.com/AlexeyAB/darknet.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
930 B
34 lines
930 B
#ifndef GEMM_H |
|
#define GEMM_H |
|
|
|
void gemm_bin(int M, int N, int K, float ALPHA, |
|
char *A, int lda, |
|
float *B, int ldb, |
|
float *C, int ldc); |
|
|
|
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); |
|
|
|
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); |
|
|
|
#ifdef GPU |
|
void gemm_ongpu(int TA, int TB, int M, int N, int K, float ALPHA, |
|
float *A_gpu, int lda, |
|
float *B_gpu, int ldb, |
|
float BETA, |
|
float *C_gpu, int ldc); |
|
|
|
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); |
|
#endif |
|
#endif
|
|
|