| | |
| | | #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; |
| | |
| | | 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); |
| | | |
| | | |