| | |
|---|
| | | #include <stddef.h> |
|---|
| | | |
|---|
| | | void convolution_2d(int w, int h, int ksize, int n, int c, int pad, int stride, |
|---|
| | | float *weights, float *input, float *output); |
|---|
| | | float *weights, float *input, float *output, float *mean); |
|---|
| | | |
|---|
| | | static inline void set_bit(unsigned char *const dst, size_t index) { |
|---|
| | | size_t dst_i = index / 8; |
|---|
| | |
|---|
| | | float *B, int ldb, |
|---|
| | | float *C, int ldc); |
|---|
| | | |
|---|
| | | |
|---|
| | | void forward_maxpool_layer_avx(float *src, float *dst, int *indexes, int size, int w, int h, int out_w, int out_h, int c, |
|---|
| | | int pad, int stride, int batch); |
|---|
| | | |
|---|
| | | |
|---|
| | | void gemm(int TA, int TB, int M, int N, int K, float ALPHA, |
|---|
| | | float *A, int lda, |
|---|
| | | float *B, int ldb, |
|---|
