1 write to norm
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaModel.cs (1)
40
this.
norm
= new RMSNorm(config.HiddenSize, config.RmsNormEps, config.DType);
1 reference to norm
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaModel.cs (1)
146
hiddenStates = this.
norm
.forward(hiddenStates);