5 references to PrevKeyKey
Microsoft.ML.TorchSharp (5)
NasBert\Modules\MultiHeadAttention.cs (5)
182if (savedState.ContainsKey(PrevKeyKey) && savedState.ContainsKey(PrevValueKey) && staticKv) 219if (savedState.ContainsKey(PrevKeyKey)) 221var prevKey = savedState[PrevKeyKey].view(batchSize * _numHeads, -1, _headDim); 235savedState[PrevKeyKey].Dispose(); 236savedState[PrevKeyKey] = k?.view(batchSize, _numHeads, -1, _headDim);