4 references to CausalLMModelOutput
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaModel.cs (1)
152
return new
CausalLMModelOutput
(lastHiddenState: hiddenStates, allHiddenStates: allHiddenStates.ToArray(), attentions: allAttentions.ToArray(), cache: this._cache);
Microsoft.ML.GenAI.Mistral (1)
MistralModel.cs (1)
146
return new
CausalLMModelOutput
(lastHiddenState: hiddenStates, allHiddenStates: allHiddenStates.ToArray(), attentions: allAttentions.ToArray(), cache: this._cache);
Microsoft.ML.GenAI.Phi (2)
Module\Phi3Model.cs (1)
145
return new
CausalLMModelOutput
(lastHiddenState: hiddenStates, allHiddenStates: allHiddenStates.ToArray(), attentions: allAttentions.ToArray(), cache: this._cache);
Phi2\Phi2ForCasualLM.cs (1)
47
return new
CausalLMModelOutput
(lastHiddenState: hiddenState, logits: lmLogits);