3 writes to _cache
Microsoft.ML.GenAI.LLaMA (3)
Module\LlamaModel.cs (3)
41
this.
_cache
= new DynamicKVCache();
58
this.
_cache
= input.OverrideCache;
62
this.
_cache
= null;
2 references to _cache
Microsoft.ML.GenAI.LLaMA (2)
Module\LlamaModel.cs (2)
135
pastKeyValue: this.
_cache
,
152
return new CausalLMModelOutput(lastHiddenState: hiddenStates, allHiddenStates: allHiddenStates.ToArray(), attentions: allAttentions.ToArray(), cache: this.
_cache
);