darknet  v3
im2col.h
Go to the documentation of this file.
1 #ifndef IM2COL_H
2 #define IM2COL_H
3 
4 void im2col_cpu(float* data_im,
5  int channels, int height, int width,
6  int ksize, int stride, int pad, float* data_col);
7 
8 #ifdef GPU
9 
10 void im2col_gpu(float *im,
11  int channels, int height, int width,
12  int ksize, int stride, int pad,float *data_col);
13 
14 #endif
15 #endif
im2col_gpu
void im2col_gpu(float *im, int channels, int height, int width, int ksize, int stride, int pad, float *data_col)
Definition: im2col_kernels.cu:48
im2col_cpu
void im2col_cpu(float *data_im, int channels, int height, int width, int ksize, int stride, int pad, float *data_col)
Definition: im2col.c:16
Generated by   doxygen 1.8.13