grenade/cbits/im2col.h

24 lines
934 B
C
Raw Normal View History

2016-12-12 15:35:00 +03:00
#include <stdio.h>
#include <stdint.h>
2016-12-12 17:11:18 +03:00
#include <string.h>
2016-12-12 15:35:00 +03:00
void im2col_cpu(const double* data_im, const int channels,
2016-12-12 15:35:00 +03:00
const int height, const int width, const int kernel_h, const int kernel_w,
const int stride_h, const int stride_w,
double* data_col);
2016-12-12 17:11:18 +03:00
void col2im_cpu(const double* data_col, const int channels,
2016-12-12 17:11:18 +03:00
const int height, const int width, const int kernel_h, const int kernel_w,
const int stride_h, const int stride_w,
double* data_im);
void pool_forwards_cpu(const double* data_im, const int channels,
const int height, const int width, const int kernel_h, const int kernel_w,
const int stride_h, const int stride_w,
double* data_pooled);
void pool_backwards_cpu(const double* data_im, const double* data_pooled,
const int channels, const int height, const int width, const int kernel_h,
const int kernel_w, const int stride_h, const int stride_w,
double* data_backgrad );