| 175 | |
| 176 | template <typename T> |
| 177 | void im2col(const SimpleTensor<T> &src, |
| 178 | SimpleTensor<T> &dst, |
| 179 | const Size2D &kernel_dims, |
| 180 | const PadStrideInfo &conv_info, |
| 181 | bool has_bias, |
| 182 | unsigned int num_groups, |
| 183 | unsigned int channel_pad_right) |
| 184 | { |
| 185 | switch (src.data_layout()) |
| 186 | { |
| 187 | case DataLayout::NCHW: |
| 188 | { |
| 189 | im2col_nchw(src, dst, kernel_dims, conv_info, has_bias, num_groups, channel_pad_right); |
| 190 | break; |
| 191 | } |
| 192 | case DataLayout::NHWC: |
| 193 | { |
| 194 | im2col_nhwc(src, dst, kernel_dims, conv_info, has_bias, channel_pad_right); |
| 195 | break; |
| 196 | } |
| 197 | default: |
| 198 | { |
| 199 | ARM_COMPUTE_ERROR("Not supported."); |
| 200 | break; |
| 201 | } |
| 202 | } |
| 203 | } |
| 204 | |
| 205 | template void im2col(const SimpleTensor<uint8_t> &src, |
| 206 | SimpleTensor<uint8_t> &dst, |
no test coverage detected