2 references to MakeCasualMask
Microsoft.ML.GenAI.Core (2)
Utility\AttentionMaskConverter.cs (2)
54
casual4dMask =
MakeCasualMask
(inputShape, dType, attentionMask2d.device, pastKeyValuesLength, this._slidingWindow);
91
causal4DMask =
MakeCasualMask
(inputShape, dType, device, pastKeyValueLength, this._slidingWindow);