| 64 | } |
| 65 | |
| 66 | array convolve2GradientNN( |
| 67 | const array &incoming_gradient, const array &original_signal, |
| 68 | const array &original_filter, const array &convolved_output, |
| 69 | const dim4 stride, // NOLINT(performance-unnecessary-value-param) |
| 70 | const dim4 padding, // NOLINT(performance-unnecessary-value-param) |
| 71 | const dim4 dilation, // NOLINT(performance-unnecessary-value-param) |
| 72 | af_conv_gradient_type gradType) { |
| 73 | af_array out = 0; |
| 74 | AF_THROW(af_convolve2_gradient_nn( |
| 75 | &out, incoming_gradient.get(), original_signal.get(), |
| 76 | original_filter.get(), convolved_output.get(), 2, stride.get(), 2, |
| 77 | padding.get(), 2, dilation.get(), gradType)); |
| 78 | return array(out); |
| 79 | } |
| 80 | |
| 81 | array convolve3(const array &signal, const array &filter, const convMode mode, |
| 82 | convDomain domain) { |
no test coverage detected