00001 #ifndef IM2COL_H 00002 #define IM2COL_H 00003 00004 void im2col_cpu(float* data_im, 00005 int channels, int height, int width, 00006 int ksize, int stride, int pad, float* data_col); 00007 00008 #ifdef GPU 00009 00010 void im2col_ongpu(float *im, 00011 int channels, int height, int width, 00012 int ksize, int stride, int pad,float *data_col); 00013 00014 #endif 00015 #endif