2 implementations of FallbackToCpu
Microsoft.ML.Core (1)
Environment\HostEnvironmentBase.cs (1)
335
public bool
FallbackToCpu
{ get; set; }
Microsoft.ML.Data (1)
MLContext.cs (1)
102
public bool
FallbackToCpu
1 write to FallbackToCpu
Microsoft.ML.Core (1)
Environment\HostEnvironmentBase.cs (1)
428
((IHostEnvironmentInternal)host).
FallbackToCpu
= FallbackToCpu;
3 references to FallbackToCpu
Microsoft.ML.OnnxTransformer (2)
OnnxCatalog.cs (1)
505
fallbackToCpu = localEnvironment.
FallbackToCpu
;
OnnxTransform.cs (1)
225
options.FallbackToCpu = localEnvironment.
FallbackToCpu
;
Microsoft.ML.TorchSharp (1)
Utils\TorchUtils.cs (1)
40
if (((IHostEnvironmentInternal)env).
FallbackToCpu
== false && device == CPU && ((IHostEnvironmentInternal)env).GpuDeviceId != null)