_cache

2 writes to _cache

Microsoft.ML.GenAI.LLaMA (2)

Module\LlamaModel.cs (2)

47this._cache = new DynamicKVCache(); 62this._cache = input.OverrideCache;

2 references to _cache

Microsoft.ML.GenAI.LLaMA (2)

Module\LlamaModel.cs (2)

135pastKeyValue: this._cache, 152return new CausalLMModelOutput(lastHiddenState: hiddenStates, allHiddenStates: allHiddenStates.ToArray(), attentions: allAttentions.ToArray(), cache: this._cache);