AlexeyAB
2018-08-10 a9fef1bd66e6b2c40c344c1bdcd33bb1d209461c
src/gemm.h
@@ -4,6 +4,9 @@
#include <stdint.h>
#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);
static inline void set_bit(unsigned char *const dst, size_t index) {
    size_t dst_i = index / 8;
    int dst_shift = index % 8;
@@ -31,6 +34,10 @@
    int channels, int height, int width,
    int ksize, int stride, int pad, float* data_col);
void im2col_cpu_custom_transpose(float* data_im,
    int channels, int height, int width,
    int ksize, int stride, int pad, float* data_col, int ldb_align);
void activate_array_cpu_custom(float *x, const int n, const ACTIVATION a);