9 references to InitializeDevice
Microsoft.ML.TorchSharp (9)
AutoFormerV2\ObjectDetectionTrainer.cs (4)
213Device = TorchUtils.InitializeDevice(Parent.Host); 226Device = TorchUtils.InitializeDevice(Parent.Host); 524Device = TorchUtils.InitializeDevice(env); 692device: TorchUtils.InitializeDevice(env));
Roberta\QATrainer.cs (3)
201Device = TorchUtils.InitializeDevice(Parent.Host); 209Device = TorchUtils.InitializeDevice(Parent.Host); 575Device = TorchUtils.InitializeDevice(env);
TorchSharpBaseTrainer.cs (2)
159Device = TorchUtils.InitializeDevice(Parent.Host); 391Device = TorchUtils.InitializeDevice(env);