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