12 references to ApplyOnnxModel
Microsoft.ML.OnnxTransformerTest (7)
OnnxTransformTests.cs (7)
158
ML.Transforms.
ApplyOnnxModel
(options.OutputColumns, options.InputColumns, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
196
var est = ML.Transforms.
ApplyOnnxModel
(outputNames, inputNames, modelFile, gpuDeviceId, fallbackToCpu);
436
var pipeline = ML.Transforms.
ApplyOnnxModel
(new[] { "outa", "outb" }, new[] { "ina", "inb" }, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
476
var pipeline = ML.Transforms.
ApplyOnnxModel
(new[] { "outb", "outa" }, new[] { "ina", "inb" }, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
502
pipeline = ML.Transforms.
ApplyOnnxModel
(new[] { "outb" }, new[] { "ina", "inb" }, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
694
var pipeline = ML.Transforms.
ApplyOnnxModel
(new[] { "output" }, new[] { "input" }, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
747
var pipeline = ML.Transforms.
ApplyOnnxModel
(new[] { "output" }, new[] { "input" }, modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
Microsoft.ML.Tests (5)
OnnxConversionTest.cs (3)
1632
var onnxEstimator = mlContext.Transforms.
ApplyOnnxModel
(outputNames, inputNames, onnxModelPath, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
1806
var onnxEstimator = mlContext.Transforms.
ApplyOnnxModel
(outputNames, inputNames, onnxModelPath, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
1819
var onnxEstimator2 = mlContext.Transforms.
ApplyOnnxModel
(outputNames, inputNames, onnxModelPath2, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
OnnxSequenceTypeWithAttributesTest.cs (2)
46
var pipeline = ctx.Transforms.
ApplyOnnxModel
(
85
var pipeline = ctx.Transforms.
ApplyOnnxModel
(