1 write to _rotaryEmb
Microsoft.ML.GenAI.Mistral (1)
MistralModel.cs (1)
43
this.
_rotaryEmb
= config.RopeScaling switch
1 reference to _rotaryEmb
Microsoft.ML.GenAI.Mistral (1)
MistralModel.cs (1)
117
var embOutput = this.
_rotaryEmb
.forward(new RotaryEmbeddingInput(hiddenStates, positionIds, pastKeyValuesLength));