3 writes to Model
Microsoft.ML.OnnxTransformer (3)
OnnxTransform.cs (3)
265
Model
= new OnnxModel(env, options.ModelFile, options.GpuDeviceId, options.FallbackToCpu, ownModelFile: false, shapeDictionary: shapeDictionary, options.RecursionLimit,
273
Model
= OnnxModel.CreateFromStream(options.ModelBytes, env, options.GpuDeviceId, options.FallbackToCpu, shapeDictionary: shapeDictionary, options.RecursionLimit);
280
Model
= OnnxModel.CreateFromBytes(modelBytes, env, options.GpuDeviceId, options.FallbackToCpu, shapeDictionary: shapeDictionary, options.RecursionLimit);
17 references to Model
Microsoft.ML.OnnxTransformer (17)
OnnxTransform.cs (17)
288
var modelInfo =
Model
.ModelInfo;
289
Inputs = (options.InputColumns.Count() == 0) ?
Model
.ModelInfo.InputNames.ToArray() : options.InputColumns;
290
Outputs = (options.OutputColumns.Count() == 0) ?
Model
.ModelInfo.OutputNames.ToArray() : options.OutputColumns;
292
var numModelOutputs =
Model
.ModelInfo.OutputsInfo.Length;
295
var outputInfo =
Model
.ModelInfo.GetOutput(Outputs[i]);
424
ctx.SaveBinaryStream("OnnxModel", w => { w.WriteByteArray(File.ReadAllBytes(
Model
.ModelStream.Name)); });
477
return
Model
.ModelInfo.OutputNames.IndexOf(Outputs[iinfo]);
485
Model
?.Dispose();
515
var model = _parent.
Model
;
585
var graph = _parent.
Model
.Graph;
624
if (_parent.
Model
.ModelInfo.OutputsInfo[_parent.MapDataViewColumnToOnnxOutputTensor(iinfo)].DataViewType is VectorDataViewType vectorType)
635
var type = _parent.
Model
.ModelInfo.OutputsInfo[_parent.MapDataViewColumnToOnnxOutputTensor(iinfo)].DataViewType.RawType;
697
outputCache.OutputOnnxValues = _parent.
Model
.Run(inputNameOnnxValues, activeOutputColNames);
762
var caster = _parent.
Model
.ModelInfo.OutputsInfo[_parent.MapDataViewColumnToOnnxOutputTensor(iinfo)].Caster;
1065
if (Transformer.
Model
.ModelInfo.InputsInfo[i].Shape.Where(x => x == 0).Count() > 1)
1072
var inputsInfo = Transformer.
Model
.ModelInfo.InputsInfo;
1073
var idx = Transformer.
Model
.ModelInfo.InputNames.IndexOf(input);