2 writes to Model
Microsoft.ML.OnnxTransformer (2)
OnnxTransform.cs (2)
260
Model
= new OnnxModel(options.ModelFile, options.GpuDeviceId, options.FallbackToCpu, ownModelFile: false, shapeDictionary: shapeDictionary, options.RecursionLimit,
267
Model
= OnnxModel.CreateFromBytes(modelBytes, env, options.GpuDeviceId, options.FallbackToCpu, shapeDictionary: shapeDictionary, options.RecursionLimit);
17 references to Model
Microsoft.ML.OnnxTransformer (17)
OnnxTransform.cs (17)
275
var modelInfo =
Model
.ModelInfo;
276
Inputs = (options.InputColumns.Count() == 0) ?
Model
.ModelInfo.InputNames.ToArray() : options.InputColumns;
277
Outputs = (options.OutputColumns.Count() == 0) ?
Model
.ModelInfo.OutputNames.ToArray() : options.OutputColumns;
279
var numModelOutputs =
Model
.ModelInfo.OutputsInfo.Length;
282
var outputInfo =
Model
.ModelInfo.GetOutput(Outputs[i]);
353
ctx.SaveBinaryStream("OnnxModel", w => { w.WriteByteArray(File.ReadAllBytes(
Model
.ModelStream.Name)); });
406
return
Model
.ModelInfo.OutputNames.IndexOf(Outputs[iinfo]);
414
Model
?.Dispose();
444
var model = _parent.
Model
;
514
var graph = _parent.
Model
.Graph;
553
if (_parent.
Model
.ModelInfo.OutputsInfo[_parent.MapDataViewColumnToOnnxOutputTensor(iinfo)].DataViewType is VectorDataViewType vectorType)
564
var type = _parent.
Model
.ModelInfo.OutputsInfo[_parent.MapDataViewColumnToOnnxOutputTensor(iinfo)].DataViewType.RawType;
626
outputCache.OutputOnnxValues = _parent.
Model
.Run(inputNameOnnxValues, activeOutputColNames);
691
var caster = _parent.
Model
.ModelInfo.OutputsInfo[_parent.MapDataViewColumnToOnnxOutputTensor(iinfo)].Caster;
955
if (Transformer.
Model
.ModelInfo.InputsInfo[i].Shape.Where(x => x == 0).Count() > 1)
962
var inputsInfo = Transformer.
Model
.ModelInfo.InputsInfo;
963
var idx = Transformer.
Model
.ModelInfo.InputNames.IndexOf(input);