3 references to RotaryEmbeddingInput
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaModel.cs (1)
123
var embOutput = this._rotaryEmb.forward(new
RotaryEmbeddingInput
(hiddenStates, positionIds, pastKeyValuesLength));
Microsoft.ML.GenAI.Mistral (1)
MistralModel.cs (1)
117
var embOutput = this._rotaryEmb.forward(new
RotaryEmbeddingInput
(hiddenStates, positionIds, pastKeyValuesLength));
Microsoft.ML.GenAI.Phi (1)
Module\Phi3Model.cs (1)
115
var positionEmbeddings = this._rotaryEmb.forward(new
RotaryEmbeddingInput
(hiddenStates, positionIds, seqLength));