AlexeyAB
2018-08-16 c2a8cc2ad3f4b31243cbf577440ee45bafcc7f92
src/gemm.h
@@ -1,6 +1,11 @@
#ifndef GEMM_H
#define GEMM_H
#include "activations.h"
#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, float *mean);
static inline void set_bit(unsigned char *const dst, size_t index) {
    size_t dst_i = index / 8;
@@ -29,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);