3 references to _gpuDeviceId
Microsoft.ML.IntegrationTests (3)
ONNX.cs (3)
60
var onnxEstimator = mlContext.Transforms.ApplyOnnxModel(modelPath, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
106
var onnxEstimator = mlContext.Transforms.ApplyOnnxModel(modelPath, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);
158
var onnxEstimator = mlContext.Transforms.ApplyOnnxModel(modelPath, gpuDeviceId:
_gpuDeviceId
, fallbackToCpu: _fallbackToCpu);