| 158 | } |
| 159 | |
| 160 | modules::Conv1dWeights load_conv1d( |
| 161 | core::BackendWeightStore & store, |
| 162 | const assets::TensorSource & source, |
| 163 | const std::string & prefix, |
| 164 | int64_t in_channels, |
| 165 | int64_t out_channels, |
| 166 | int64_t kernel_size, |
| 167 | assets::TensorStorageType weight_storage_type) { |
| 168 | modules::Conv1dWeights weights; |
| 169 | weights.weight = store.load_tensor(source, prefix + ".weight", weight_storage_type, {out_channels, in_channels, kernel_size}); |
| 170 | weights.bias = store.load_tensor(source, prefix + ".bias", assets::TensorStorageType::F32, {out_channels}); |
| 171 | return weights; |
| 172 | } |
| 173 | |
| 174 | modules::DepthwiseConv1dWeights load_depthwise_conv1d( |
| 175 | core::BackendWeightStore & store, |
no test coverage detected