| 190 | }; |
| 191 | |
| 192 | core::TensorValue conv1d( |
| 193 | core::ModuleBuildContext & ctx, |
| 194 | const core::TensorValue & input, |
| 195 | const Conv1dLayer & layer, |
| 196 | int64_t padding, |
| 197 | int64_t dilation) { |
| 198 | return modules::Conv1dModule({ |
| 199 | layer.in_channels, |
| 200 | layer.out_channels, |
| 201 | layer.kernel, |
| 202 | 1, |
| 203 | static_cast<int>(padding), |
| 204 | static_cast<int>(dilation), |
| 205 | true, |
| 206 | }).build(ctx, input, layer.conv); |
| 207 | } |
| 208 | |
| 209 | core::TensorValue replicate_pad1d( |
| 210 | core::ModuleBuildContext & ctx, |
no test coverage detected