6 references to ApplyOnnxModel
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",