| 15 | }; |
| 16 | |
| 17 | class LayerScaleModule { |
| 18 | public: |
| 19 | core::TensorValue build( |
| 20 | core::ModuleBuildContext & ctx, |
| 21 | const core::TensorValue & input, |
| 22 | const LayerScaleWeights & weights) const; |
| 23 | }; |
| 24 | |
| 25 | struct AdaLNModulationWeights { |
| 26 | LinearWeights projection; |