1 write to Device
Microsoft.ML.TorchSharp (1)
TorchSharpBaseTrainer.cs (1)
391
Device
= TorchUtils.InitializeDevice(env);
2 references to Device
Microsoft.ML.TorchSharp (2)
NasBert\NasBertTrainer.cs (1)
604
var inputTensor = torch.tensor(tokens, device: Parent.
Device
);
TorchSharpBaseTrainer.cs (1)
399
if (
Device
.type == DeviceType.CUDA)