1 write to Attentions
Microsoft.ML.GenAI.Core (1)
Module\Attention.cs (1)
57
this.
Attentions
= attentions;
3 references to Attentions
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaDecoderLayer.cs (1)
151
attentions: input.OutputAttentions ? selfAttnOutput.
Attentions
?.MoveToOuterDisposeScope() : null,
Microsoft.ML.GenAI.Mistral (1)
MistralDecoderLayer.cs (1)
145
attentions: input.OutputAttentions ? selfAttnOutput.
Attentions
?.MoveToOuterDisposeScope() : null,
Microsoft.ML.GenAI.Phi (1)
Module\Phi3DecoderLayer.cs (1)
126
var selfAttnWeights = output.
Attentions
;