| 220 | |
| 221 | template<typename T> |
| 222 | Array<T> convolve2(Array<T> const &signal, Array<T> const &filter, |
| 223 | const dim4 stride, const dim4 padding, const dim4 dilation) { |
| 224 | #ifdef WITH_CUDNN |
| 225 | if (getCudnnPlugin().isLoaded()) { |
| 226 | checkTypeSupport<T>(); |
| 227 | return convolve2_cudnn<T>(signal, filter, stride, padding, dilation); |
| 228 | } |
| 229 | #endif |
| 230 | return convolve2_base<T>(signal, filter, stride, padding, dilation); |
| 231 | } |
| 232 | |
| 233 | #define INSTANTIATE(T) \ |
| 234 | template Array<T> convolve2<T>(Array<T> const &signal, \ |