CUDA so fast

This commit is contained in:
Joseph Redmon
2015-01-22 16:38:24 -08:00
parent 4ac78c8926
commit 809f924db2
57 changed files with 1116 additions and 2181 deletions

13
src/col2im.h Normal file
View File

@ -0,0 +1,13 @@
#ifndef COL2IM_H
#define COL2IM_H
void col2im_cpu(float* data_col,
int channels, int height, int width,
int ksize, int stride, int pad, float* data_im);
#ifdef GPU
void col2im_ongpu(float *data_col, int batch,
int channels, int height, int width,
int ksize, int stride, int pad, float *data_im);
#endif
#endif