mirror of https://github.com/pjreddie/darknet.git
16 lines
329 B
C
16 lines
329 B
C
#ifndef IM2COL_H
|
|
#define IM2COL_H
|
|
|
|
void im2col_cpu(float* data_im,
|
|
int channels, int height, int width,
|
|
int ksize, int stride, int pad, float* data_col);
|
|
|
|
#ifdef GPU
|
|
|
|
void im2col_gpu(float *im,
|
|
int channels, int height, int width,
|
|
int ksize, int stride, int pad,float *data_col);
|
|
|
|
#endif
|
|
#endif
|