10 references to TorchUtils
Microsoft.ML.TorchSharp (10)
AutoFormerV2\ObjectDetectionTrainer.cs (4)
213Device = TorchUtils.InitializeDevice(Parent.Host); 226Device = TorchUtils.InitializeDevice(Parent.Host); 524Device = TorchUtils.InitializeDevice(env); 692device: TorchUtils.InitializeDevice(env));
NasBert\Modules\IncrementalState.cs (1)
85TorchUtils.DisposeDictionaryWithTensor(oldState);
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);