Esempi in C++ (Cpp) per im2col_nd_core_cpu

Esempio n. 1

File: im2col.cpp Progetto: dtmoodie/caffe

void col2im_nd_cpu(const Dtype* data_col, const int num_spatial_axes,
    const int* im_shape, const int* col_shape,
    const int* kernel_shape, const int* pad, const int* stride,
    const int* dilation, Dtype* data_im) {
  const bool kIm2Col = false;
  im2col_nd_core_cpu(data_col, kIm2Col, num_spatial_axes, im_shape, col_shape,
                     kernel_shape, pad, stride, dilation, data_im);
}

Esempio n. 2

File: im2col.cpp Progetto: mfigurnov/perforated-cnn-caffe

void im2col_nd_cpu(const Dtype* data_im, const int num_spatial_axes,
                   const int* im_shape, const int* col_shape,
                   const int* kernel_shape, const int* pad, const int* stride,
                   Dtype* data_col) {
    const bool kIm2Col = true;
    im2col_nd_core_cpu(data_im, kIm2Col, num_spatial_axes, im_shape, col_shape,
                       kernel_shape, pad, stride, data_col);
}

Esempio n. 3

File: im2col.cpp Progetto: naibaf7/caffe

void im2col_nd_cpu(const Dtype* data_im, const int_tp num_spatial_axes,
                   const int_tp* im_shape, const int_tp* col_shape,
                   const int_tp* kernel_shape, const int_tp* pad,
                   const int_tp* stride, const int_tp* dilation,
                   Dtype* data_col, const QuantizerValues* const data_quant) {
  const bool kIm2Col = true;
  im2col_nd_core_cpu(data_im, kIm2Col, num_spatial_axes, im_shape, col_shape,
                     kernel_shape, pad, stride, dilation, data_col, data_quant);
}