| 350 | |
| 351 | template<typename T> |
| 352 | Array<T> conv2DataGradient(const Array<T> &incoming_gradient, |
| 353 | const Array<T> &original_signal, |
| 354 | const Array<T> &original_filter, |
| 355 | const Array<T> &convolved_output, af::dim4 stride, |
| 356 | af::dim4 padding, af::dim4 dilation) { |
| 357 | #ifdef WITH_CUDNN |
| 358 | if (getCudnnPlugin().isLoaded()) { |
| 359 | checkTypeSupport<T>(); |
| 360 | return data_gradient_cudnn<T>(incoming_gradient, original_signal, |
| 361 | original_filter, convolved_output, stride, |
| 362 | padding, dilation); |
| 363 | } |
| 364 | #endif |
| 365 | return data_gradient_base<T>(incoming_gradient, original_signal, |
| 366 | original_filter, convolved_output, stride, |
| 367 | padding, dilation); |
| 368 | } |
| 369 | |
| 370 | template<typename T> |
| 371 | Array<T> filter_gradient_base(const Array<T> &incoming_gradient, |