7 references to ApplyOnnxModel
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);
Microsoft.ML.OnnxTransformerTest (5)
OnnxTransformTests.cs (5)
308
.Append(ML.Transforms.
ApplyOnnxModel
("softmaxout_1", "data_0", modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu));
369
var pipeline = ML.Transforms.
ApplyOnnxModel
("softmaxout_1", "data_0", modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
408
var pipeline = ML.Transforms.
ApplyOnnxModel
("softmaxout_1", "data_0", modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);
417
Assert.Throws<InvalidOperationException>(() => ML.Transforms.
ApplyOnnxModel
("softmaxout_1", "data_0", modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu));
644
.Append(ML.Transforms.
ApplyOnnxModel
("softmaxout_1", "data_0", modelFile, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu)); // Map column "data_0" to column "softmaxout_1"
Microsoft.ML.Samples (1)
Dynamic\Transforms\ApplyONNXModelWithInMemoryImages.cs (1)
46
.Append(mlContext.Transforms.
ApplyOnnxModel
("softmaxout_1",