1 reference to ExpandMask
Microsoft.ML.GenAI.Core (1)
Utility\AttentionMaskConverter.cs (1)
61
var expandedAttnMask =
ExpandMask
(attentionMask2d, dType, queryLength).to(attentionMask2d.device);