| 56 | |
| 57 | S3FlowEncoderWeights::Conv1dWeights load_flow_conv1d( |
| 58 | engine::core::BackendWeightStore & store, |
| 59 | const engine::assets::TensorSource & source, |
| 60 | const std::string & prefix, |
| 61 | int64_t out_channels, |
| 62 | int64_t in_channels, |
| 63 | int64_t kernel, |
| 64 | int64_t stride, |
| 65 | int64_t padding, |
| 66 | engine::assets::TensorStorageType weight_storage_type) { |
| 67 | S3FlowEncoderWeights::Conv1dWeights conv; |
| 68 | conv.out_channels = out_channels; |
| 69 | conv.in_channels = in_channels; |
| 70 | conv.kernel = kernel; |
| 71 | conv.stride = stride; |
| 72 | conv.padding = padding; |
| 73 | conv.weight_tensor = store.load_tensor(source, prefix + ".weight", weight_storage_type, {out_channels, in_channels, kernel}); |
| 74 | conv.bias_tensor = store.load_f32_tensor(source, prefix + ".bias", {out_channels}); |
| 75 | return conv; |
| 76 | } |
| 77 | |
| 78 | S3FlowDecoderWeights::LayerNormWeights load_decoder_layer_norm( |
| 79 | engine::core::BackendWeightStore & store, |
| 80 | const engine::assets::TensorSource & source, |
no test coverage detected