1 write to lm_head
Microsoft.ML.GenAI.LLaMA (1)
LlamaForCausalLM.cs (1)
33
lm_head
= new GenAILinear(config.HiddenSize, config.VocabSize, hasBias: false);
1 reference to lm_head
Microsoft.ML.GenAI.LLaMA (1)
LlamaForCausalLM.cs (1)
43
var logits = this.
lm_head
.forward(outputs.LastHiddenState);