1 #ifndef _CAFFE_UTIL_IM2COL_HPP_ 2 #define _CAFFE_UTIL_IM2COL_HPP_ 6 template <
typename Dtype>
7 void im2col_nd_cpu(
const Dtype* data_im,
const int num_spatial_axes,
8 const int* im_shape,
const int* col_shape,
9 const int* kernel_shape,
const int* pad,
const int* stride,
10 const int* dilation, Dtype* data_col);
12 template <
typename Dtype>
13 void im2col_cpu(
const Dtype* data_im,
const int channels,
14 const int height,
const int width,
const int kernel_h,
const int kernel_w,
15 const int pad_h,
const int pad_w,
const int stride_h,
16 const int stride_w,
const int dilation_h,
const int dilation_w,
19 template <
typename Dtype>
20 void col2im_nd_cpu(
const Dtype* data_col,
const int num_spatial_axes,
21 const int* im_shape,
const int* col_shape,
22 const int* kernel_shape,
const int* pad,
const int* stride,
23 const int* dilation, Dtype* data_im);
25 template <
typename Dtype>
26 void col2im_cpu(
const Dtype* data_col,
const int channels,
27 const int height,
const int width,
const int kernel_h,
const int kernel_w,
28 const int pad_h,
const int pad_w,
const int stride_h,
29 const int stride_w,
const int dilation_h,
const int dilation_w,
32 template <
typename Dtype>
33 void im2col_nd_gpu(
const Dtype* data_im,
const int num_spatial_axes,
34 const int col_size,
const int* im_shape,
const int* col_shape,
35 const int* kernel_shape,
const int* pad,
const int* stride,
36 const int* dilation, Dtype* data_col);
38 template <
typename Dtype>
39 void im2col_gpu(
const Dtype* data_im,
const int channels,
40 const int height,
const int width,
const int kernel_h,
const int kernel_w,
41 const int pad_h,
const int pad_w,
const int stride_h,
42 const int stride_w,
const int dilation_h,
const int dilation_w,
45 template <
typename Dtype>
46 void col2im_nd_gpu(
const Dtype* data_col,
const int num_spatial_axes,
47 const int im_size,
const int* im_shape,
const int* col_shape,
48 const int* kernel_shape,
const int* pad,
const int* stride,
49 const int* dilation, Dtype* data_im);
51 template <
typename Dtype>
52 void col2im_gpu(
const Dtype* data_col,
const int channels,
53 const int height,
const int width,
const int kernel_h,
const int kernel_w,
54 const int pad_h,
const int pad_w,
const int stride_h,
55 const int stride_w,
const int dilation_h,
const int dilation_w,
60 #endif // CAFFE_UTIL_IM2COL_HPP_ A layer factory that allows one to register layers. During runtime, registered layers can be called b...
Definition: blob.hpp:14