1 write to InputIds
Microsoft.ML.GenAI.Core (1)
Pipeline\CausalLMModelInput.cs (1)
33
this.
InputIds
= inputIds;
10 references to InputIds
Microsoft.ML.GenAI.Core (3)
Trainer\CasualLMSupervisedFineTuningTrainer.cs (3)
52
var maxLen = batch.Max(x => x.
InputIds
.size(1));
54
var inputIds = torch.cat(batch.Select(x => nn.functional.pad(x.
InputIds
, [0, maxLen - x.
InputIds
.shape[1]])).ToArray(), 0).to(device);
Microsoft.ML.GenAI.Core.Tests (3)
CasualLMDatasetTest.cs (3)
46
dataset[0].
InputIds
!.data<long>().Should().BeEquivalentTo([1, 2, 3, 4, 5]);
74
item.Labels!.shape.Should().BeEquivalentTo(item.
InputIds
!.shape);
75
item.AttentionMask!.shape.Should().BeEquivalentTo(item.
InputIds
!.shape);
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaModel.cs (1)
69
var inputIds = input.
InputIds
;
Microsoft.ML.GenAI.Mistral (1)
MistralModel.cs (1)
63
var inputIds = input.
InputIds
;
Microsoft.ML.GenAI.Phi (2)
Module\Phi3Model.cs (1)
66
var inputIds = input.
InputIds
;
Phi2\Phi2ForCasualLM.cs (1)
36
var inputIds = input.
InputIds
;