1 write to OutputAttentions
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaDecoderLayer.cs (1)
30this.OutputAttentions = outputAttentions;
2 references to OutputAttentions
Microsoft.ML.GenAI.LLaMA (2)
Module\LlamaDecoderLayer.cs (2)
132outputAttentions: input.OutputAttentions); 151attentions: input.OutputAttentions ? selfAttnOutput.Attentions?.MoveToOuterDisposeScope() : null,