3 instantiations of RotaryEmbeddingInput
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaModel.cs (1)
123var embOutput = this._rotaryEmb.forward(new RotaryEmbeddingInput(hiddenStates, positionIds, pastKeyValuesLength));
Microsoft.ML.GenAI.Mistral (1)
MistralModel.cs (1)
117var embOutput = this._rotaryEmb.forward(new RotaryEmbeddingInput(hiddenStates, positionIds, pastKeyValuesLength));
Microsoft.ML.GenAI.Phi (1)
Module\Phi3Model.cs (1)
115var positionEmbeddings = this._rotaryEmb.forward(new RotaryEmbeddingInput(hiddenStates, positionIds, seqLength));
6 references to RotaryEmbeddingInput
Microsoft.ML.GenAI.Core (2)
Module\RotaryEmbedding.cs (2)
71RotaryEmbeddingInput, 104public override RotaryEmbeddingOutput forward(RotaryEmbeddingInput input)
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaModel.cs (1)
23private readonly nn.Module<RotaryEmbeddingInput, RotaryEmbeddingOutput> _rotaryEmb;
Microsoft.ML.GenAI.Mistral (1)
MistralModel.cs (1)
23private readonly nn.Module<RotaryEmbeddingInput, RotaryEmbeddingOutput> _rotaryEmb;
Microsoft.ML.GenAI.Phi (2)
Module\Phi3Model.cs (1)
25private readonly nn.Module<RotaryEmbeddingInput, RotaryEmbeddingOutput> _rotaryEmb;
Module\Phi3SuScaledRotaryEmbedding.cs (1)
39public override RotaryEmbeddingOutput forward(RotaryEmbeddingInput input)