1 write to lm_head
Microsoft.ML.GenAI.Phi (1)
Phi2\Phi2ForCasualLM.cs (1)
28
this.
lm_head
= new GenAILinear(config.HiddenSize, config.VocabSize, dtype: config.Dtype);
1 reference to lm_head
Microsoft.ML.GenAI.Phi (1)
Phi2\Phi2ForCasualLM.cs (1)
45
var lmLogits = this.
lm_head
.forward(hiddenState);