3 instantiations of AttentionInput
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaDecoderLayer.cs (1)
126var selfAttnInput = new AttentionInput(
Microsoft.ML.GenAI.Mistral (1)
MistralDecoderLayer.cs (1)
120var selfAttnInput = new AttentionInput(
Microsoft.ML.GenAI.Phi (1)
Module\Phi3DecoderLayer.cs (1)
117var attentionInput = new AttentionInput(
6 references to AttentionInput
Microsoft.ML.GenAI.Core (2)
Module\Attention.cs (2)
68internal class Attention : nn.Module<AttentionInput, AttentionOutput> 129public override AttentionOutput forward(AttentionInput input)
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaDecoderLayer.cs (1)
126var selfAttnInput = new AttentionInput(
Microsoft.ML.GenAI.Mistral (1)
MistralDecoderLayer.cs (1)
120var selfAttnInput = new AttentionInput(
Microsoft.ML.GenAI.Phi (2)
Module\Phi3DecoderLayer.cs (2)
71private readonly nn.Module<AttentionInput, AttentionOutput> self_attn; 117var attentionInput = new AttentionInput(