MakeCausalMask

2 references to MakeCausalMask

Microsoft.ML.GenAI.Core (2)

Utility\AttentionMaskConverter.cs (2)

54causal4dMask = MakeCausalMask(inputShape, dType, attentionMask2d.device, pastKeyValuesLength, this._slidingWindow); 91causal4DMask = MakeCausalMask(inputShape, dType, device, pastKeyValueLength, this._slidingWindow);