2 writes to Device
Microsoft.ML.GenAI.Core (2)
Pipeline\CausalLMPipeline.cs (2)
103this.Device = device; 113this.Device = default!;
4 references to Device
Microsoft.ML.GenAI.Core (4)
Pipeline\CausalLMPipeline.cs (4)
248var inputTensor = torch.tensor(inputIds.ToArray(), dtype: ScalarType.Int64, device: this.Device).unsqueeze(0); 249var attentionMask = torch.ones_like(inputTensor, device: this.Device); 309var inputTensor = torch.tensor(inputIds.ToArray(), dtype: ScalarType.Int64, device: this.Device).unsqueeze(0); 310var attentionMask = torch.ones_like(inputTensor, device: this.Device);