1 reference to GpuDeviceId
Microsoft.ML.AutoML (1)
SweepableEstimator\Estimators\ApplyOnnx.cs (1)
11
return context.Transforms.ApplyOnnxModel(outputColumnName: param.OutputColumnName, inputColumnName: param.InputColumnName, modelFile: param.ModelFile, gpuDeviceId: param.
GpuDeviceId
, fallbackToCpu: param.FallbackToCpu);