3 instantiations of 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.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);
18 references to CausalLMModelOutput
Microsoft.ML.GenAI.Core (6)
Pipeline\CausalLMPipeline.cs (6)
19
where TModel : nn.Module<CausalLMModelInput,
CausalLMModelOutput
>
66
where TModel : nn.Module<CausalLMModelInput,
CausalLMModelOutput
>
94
nn.Module<CausalLMModelInput,
CausalLMModelOutput
> model,
114
public nn.Module<CausalLMModelInput,
CausalLMModelOutput
> Model { get; }
146
var
output = this.Model.forward(input);
155
var
output = this.Model.forward(input);
Microsoft.ML.GenAI.LLaMA (5)
LlamaForCausalLM.cs (3)
16
public class LlamaForCausalLM : nn.Module<CausalLMModelInput,
CausalLMModelOutput
>
39
public override
CausalLMModelOutput
forward(CausalLMModelInput input)
42
var
outputs = this.model.forward(input);
Module\LlamaModel.cs (2)
18
internal class LlamaModel : nn.Module<CausalLMModelInput,
CausalLMModelOutput
>
57
public override
CausalLMModelOutput
forward(CausalLMModelInput input)
Microsoft.ML.GenAI.Phi (7)
Module\Phi3Model.cs (2)
13
internal class Phi3Model : nn.Module<CausalLMModelInput,
CausalLMModelOutput
>
54
public override
CausalLMModelOutput
forward(CausalLMModelInput input)
Phi2\Phi2ForCasualLM.cs (2)
17
public class Phi2ForCasualLM : nn.Module<CausalLMModelInput,
CausalLMModelOutput
>
33
public override
CausalLMModelOutput
forward(CausalLMModelInput input) // use_cache, output_attentions, output_hidden_states
Phi3\Phi3ForCasualLM.cs (3)
21
public class Phi3ForCasualLM : nn.Module<CausalLMModelInput,
CausalLMModelOutput
>
41
public override
CausalLMModelOutput
forward(CausalLMModelInput input)
44
var
outputs = this.model.forward(input);