mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
16 lines
343 B
C
16 lines
343 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_ongpu(float *im, int offset,
|
||
|
|
int channels, int height, int width,
|
||
|
|
int ksize, int stride, int pad,float *data_col);
|
||
|
|
|
||
|
|
#endif
|
||
|
|
#endif
|