1 write to _numKeyValueGroups
Microsoft.ML.GenAI.Core (1)
Module\Attention.cs (1)
108this._numKeyValueGroups = numKeyValueGroups;
2 references to _numKeyValueGroups
Microsoft.ML.GenAI.Core (2)
Module\Attention.cs (2)
180keyStates = Utils.RepeatKV(keyStates, this._numKeyValueGroups); 181valueStates = Utils.RepeatKV(valueStates, this._numKeyValueGroups);