1 write to Attentions
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaDecoderLayer.cs (1)
55
this.
Attentions
= attentions;
2 references to Attentions
Microsoft.ML.GenAI.LLaMA (2)
Module\LlamaModel.cs (2)
140
if (outputAttentions && layerOutput.
Attentions
is not null)
142
allAttentions.Add(layerOutput.
Attentions
);