3 instantiations of OnnxVariableInfo
Microsoft.ML.OnnxTransformer (3)
OnnxUtils.cs (3)
332info = new OnnxVariableInfo(name, shapeDictionary[name].ToList(), meta.ElementType, dataViewType, caster); 339info = new OnnxVariableInfo(name, null, typeof(NamedOnnxValue), dataViewType, caster); 341info = new OnnxVariableInfo(name, meta.Dimensions.ToList(), meta.ElementType, dataViewType, caster);
14 references to OnnxVariableInfo
Microsoft.ML.OnnxTransformer (14)
OnnxTransform.cs (3)
295var outputInfo = Model.ModelInfo.GetOutput(Outputs[i]); 518var inputNodeInfo = model.ModelInfo.GetInput(_parent.Inputs[i]); 1077var inputNodeInfo = inputsInfo[idx];
OnnxUtils.cs (11)
49public OnnxVariableInfo[] InputsInfo { get; } 53public OnnxVariableInfo[] OutputsInfo { get; } 58public OnnxVariableInfo[] InitializersInfo { get; } 60public OnnxModelInfo(IEnumerable<OnnxVariableInfo> inputsInfo, IEnumerable<OnnxVariableInfo> outputsInfo, IEnumerable<OnnxVariableInfo> initializersInfo) 73public OnnxVariableInfo GetInput(string name) 91public OnnxVariableInfo GetOutput(string name) 296private List<OnnxVariableInfo> GetOnnxVariablesFromMetadata(IReadOnlyDictionary<string, NodeMetadata> nodeMetadata, 301var onnxVariableInfos = new List<OnnxVariableInfo>(); 314OnnxVariableInfo info = null;