| 119 | } |
| 120 | |
| 121 | ActivationParams load_activation( |
| 122 | core::BackendWeightStore & store, |
| 123 | const std::shared_ptr<const engine::assets::TensorSource> & source, |
| 124 | const std::string & prefix) { |
| 125 | return { |
| 126 | store.make_f32(core::TensorShape::from_dims({kFlashSrChannels}), require_channel_param(source, prefix + ".alpha")), |
| 127 | store.make_f32(core::TensorShape::from_dims({kFlashSrChannels}), require_channel_param(source, prefix + ".inv_beta")), |
| 128 | }; |
| 129 | } |
| 130 | |
| 131 | ResBlockWeights load_resblock( |
| 132 | core::BackendWeightStore & store, |
no test coverage detected