1 reference to MakeCasualAttentionMask
Microsoft.ML.GenAI.Phi (1)
Module\Phi2Model.cs (1)
113
var casual4DMask = this.
MakeCasualAttentionMask
(batchSize, queryLength, pastKeyValueLength, attentionMask.device, dtype);