| 26 | |
| 27 | S3FlowEncoderWeights::LayerNormWeights load_flow_layer_norm( |
| 28 | engine::core::BackendWeightStore & store, |
| 29 | const engine::assets::TensorSource & source, |
| 30 | const std::string & prefix, |
| 31 | int64_t channels) { |
| 32 | S3FlowEncoderWeights::LayerNormWeights weights; |
| 33 | weights.weight_tensor = store.load_f32_tensor(source, prefix + ".weight", {channels}); |
| 34 | weights.bias_tensor = store.load_f32_tensor(source, prefix + ".bias", {channels}); |
| 35 | return weights; |
| 36 | } |
| 37 | |
| 38 | S3FlowEncoderWeights::LinearWeights load_flow_linear( |
| 39 | engine::core::BackendWeightStore & store, |
| 40 | const engine::assets::TensorSource & source, |
no test coverage detected