src/caffeine/util/im2col.hpp

   1 #ifndef _CAFFEINE_UTIL__IM2COL_HPP_
   2 #define _CAFFEINE_UTIL_IM2COL_HPP_
   3
   4 namespace caffeine {
   5
   6 template <typename Dtype>
   7 void im2col_cpu(const Dtype* data_im, const int channels,
   8     const int height, const int width, const int ksize, const int stride,
   9     Dtype* data_col);
  10
  11 template <typename Dtype>
  12 void col2im_cpu(const Dtype* data_col, const int channels,
  13     const int height, const int width, const int psize, const int stride,
  14     Dtype* data_im);
  15
  16 template <typename Dtype>
  17 void im2col_gpu(const Dtype* data_im, const int channels,
  18     const int height, const int width, const int ksize, const int stride,
  19     Dtype* data_col);
  20
  21 /*
  22 template <typename Dtype>
  23 void col2im_gpu(const Dtype* data_col, const int channels,
  24     const int height, const int width, const int psize, const int stride,
  25     Dtype* data_im);
  26 */
  27 }  // namespace caffeine
  28
  29 #endif  // CAFFEINE_UTIL_IM2COL_HPP_