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