6 instantiations of OnnxTransformer
Microsoft.ML.OnnxTransformer (6)
OnnxTransform.cs (6)
152
return new
OnnxTransformer
(env, options).MakeDataTransform(input);
228
return new
OnnxTransformer
(env, options, modelBytes);
976
: this(env, new
OnnxTransformer
(env, new string[] { }, new string[] { }, modelFile, gpuDeviceId, fallbackToCpu, shapeDictionary, recursionLimit))
994
: this(env, new
OnnxTransformer
(env, new string[] { }, new string[] { }, modelBytes, gpuDeviceId, fallbackToCpu, shapeDictionary, recursionLimit))
1015
: this(env, new
OnnxTransformer
(env, outputColumnNames, inputColumnNames, modelFile, gpuDeviceId, fallbackToCpu, shapeDictionary, recursionLimit, interOpNumThreads, intraOpNumThreads))
1041
: this(env, new
OnnxTransformer
(env, outputColumnNames, inputColumnNames, modelBytes, gpuDeviceId, fallbackToCpu, shapeDictionary, recursionLimit, interOpNumThreads, intraOpNumThreads))
52 references to OnnxTransformer
Microsoft.ML.IntegrationTests (3)
ONNX.cs (3)
61
var
onnxModel = onnxEstimator.Fit(data);
107
var
onnxModel = onnxEstimator.Fit(data);
159
var
onnxModel = onnxEstimator.Fit(data);
Microsoft.ML.OnnxTransformer (26)
OnnxTransform.cs (24)
21
[assembly: LoadableClass(
OnnxTransformer
.Summary, typeof(IDataTransform), typeof(
OnnxTransformer
),
22
typeof(
OnnxTransformer
.Options), typeof(SignatureDataTransform),
OnnxTransformer
.UserName,
OnnxTransformer
.ShortName, "OnnxTransform", "OnnxScorer")]
24
[assembly: LoadableClass(
OnnxTransformer
.Summary, typeof(IDataTransform), typeof(
OnnxTransformer
),
25
null, typeof(SignatureLoadDataTransform),
OnnxTransformer
.UserName,
OnnxTransformer
.LoaderSignature)]
27
[assembly: LoadableClass(typeof(
OnnxTransformer
), null, typeof(SignatureLoadModel),
28
OnnxTransformer
.UserName,
OnnxTransformer
.LoaderSignature)]
30
[assembly: LoadableClass(typeof(IRowMapper), typeof(
OnnxTransformer
), null, typeof(SignatureLoadRowMapper),
31
OnnxTransformer
.UserName,
OnnxTransformer
.LoaderSignature)]
33
[assembly: EntryPointModule(typeof(
OnnxTransformer
))]
146
loaderAssemblyName: typeof(
OnnxTransformer
).Assembly.FullName);
160
private static
OnnxTransformer
Create(IHostEnvironment env, ModelLoadContext ctx)
236
base(Contracts.CheckRef(env, nameof(env)).Register(nameof(
OnnxTransformer
)))
491
private readonly
OnnxTransformer
_parent;
506
public Mapper(
OnnxTransformer
parent, DataViewSchema inputSchema) :
960
public sealed class OnnxScoringEstimator : TrivialEstimator<
OnnxTransformer
>
1019
internal OnnxScoringEstimator(IHostEnvironment env,
OnnxTransformer
transformer)
1020
: base(Contracts.CheckRef(env, nameof(env)).Register(nameof(
OnnxTransformer
)), transformer)
OnnxUtils.cs (2)
152
/// For example, ML.NET can shuffle the inputs so that the i-th ONNX input becomes the j-th input column of <see cref="
OnnxTransformer
"/>.
153
/// ML.NET can also only exposes a subset of ONNX outputs in <see cref="
OnnxTransformer
"/>.
Microsoft.ML.OnnxTransformerTest (12)
OnnxTransformTests.cs (12)
169
var
onnxTransformer = pipe.Fit(invalidDataWrongVectorSize);
197
var
transformer = est.Fit(dataView);
370
var
onnxTransformer = pipeline.Fit(dataView);
409
var
onnxTransformer = pipeline.Fit(dataView);
437
var
onnxTransformer = pipeline.Fit(dataView);
477
var
onnxTransformer = pipeline.Fit(dataView);
537
var
onnxTransformer = pipeline.Fit(idv);
563
var
onnxTransformer = pipeline.Fit(idv);
695
var
onnxTransformer = pipeline.Fit(dataView);
748
var
onnxTransformer = pipeline.Fit(dataView);
903
var onnxTransformer = new
OnnxTransformer
[3];
1075
var
model = pipeline.Fit(dataView);
Microsoft.ML.Samples (1)
Dynamic\ModelOperations\OnnxConversion.cs (1)
86
using
var
onnxTransformer = onnxEstimator.Fit(trainTestOriginalData.TrainSet);
Microsoft.ML.Tests (10)
OnnxConversionTest.cs (8)
827
var
onnxTransformer = onnxEstimator.Fit(data);
989
var
onnxTransformer = onnxEstimator.Fit(dataView);
1516
var
onnxTransformer = onnxEstimator.Fit(dataView);
1633
var
onnxTransformer = onnxEstimator.Fit(dataView);
1744
var
loadedModel = pipeline.Fit(dataView);
1807
var
onnxTransformer = onnxEstimator.Fit(reloadedData);
1820
var
onnxTransformer2 = onnxEstimator2.Fit(originalData);
2255
var
onnxTransformer = onnxEstimator.Fit(dataView);
OnnxSequenceTypeWithAttributesTest.cs (2)
52
var
model = pipeline.Fit(dataView);
91
var
model = pipeline.Fit(dataView);