4 writes to Logits
Microsoft.ML.GenAI.Core (1)
Pipeline\CausalLMModelOutput.cs (1)
29this.Logits = logits;
Microsoft.ML.GenAI.LLaMA (1)
LlamaForCausalLM.cs (1)
66outputs.Logits = logits;
Microsoft.ML.GenAI.Mistral (1)
MistralForCausalLM.cs (1)
45outputs.Logits = logits;
Microsoft.ML.GenAI.Phi (1)
Phi3\Phi3ForCasualLM.cs (1)
47outputs.Logits = logits;
3 references to Logits
Microsoft.ML.GenAI.Core (3)
Pipeline\CausalLMPipeline.cs (2)
151logits = output.Logits; 160logits = output.Logits?.MoveToOtherDisposeScope(inputIds) ?? throw new InvalidOperationException("Logits is null");
Trainer\CasualLMSupervisedFineTuningTrainer.cs (1)
73output.Logits!.Dispose();