@@ -29,10 +29,10 @@ internal sealed class OnnxModel
2929 /// </summary>
3030 public sealed class OnnxModelInfo
3131 {
32- public readonly OnnxNodeInfo [ ] InputsInfo ;
33- public readonly OnnxNodeInfo [ ] OutputsInfo ;
32+ public readonly OnnxVariableInfo [ ] InputsInfo ;
33+ public readonly OnnxVariableInfo [ ] OutputsInfo ;
3434
35- public OnnxModelInfo ( IEnumerable < OnnxNodeInfo > inputsInfo , IEnumerable < OnnxNodeInfo > outputsInfo )
35+ public OnnxModelInfo ( IEnumerable < OnnxVariableInfo > inputsInfo , IEnumerable < OnnxVariableInfo > outputsInfo )
3636 {
3737 InputsInfo = inputsInfo . ToArray ( ) ;
3838 OutputsInfo = outputsInfo . ToArray ( ) ;
@@ -43,7 +43,7 @@ public OnnxModelInfo(IEnumerable<OnnxNodeInfo> inputsInfo, IEnumerable<OnnxNodeI
4343 /// OnnxNodeInfo contains all the information for a given node (e.g. inputs/outputs)
4444 /// of an Onnx model.
4545 /// </summary>
46- public class OnnxNodeInfo
46+ public class OnnxVariableInfo
4747 {
4848 /// <summary>
4949 /// The Name of the node
@@ -58,7 +58,7 @@ public class OnnxNodeInfo
5858 /// </summary>
5959 public readonly System . Type Type ;
6060
61- public OnnxNodeInfo ( string name , OnnxShape shape , System . Type type )
61+ public OnnxVariableInfo ( string name , OnnxShape shape , System . Type type )
6262 {
6363 Name = name ;
6464 Shape = shape ;
@@ -163,18 +163,18 @@ public byte[] ToByteArray()
163163 /// Returns input metadata of the ONNX model.
164164 /// </summary>
165165 /// <returns>OnnxNodeInfo[]</returns>
166- private IEnumerable < OnnxNodeInfo > GetInputsInfo ( )
166+ private IEnumerable < OnnxVariableInfo > GetInputsInfo ( )
167167 {
168- return _session . InputMetadata . Select ( kv => new OnnxNodeInfo ( kv . Key , kv . Value . Dimensions . ToList ( ) , kv . Value . ElementType ) ) ;
168+ return _session . InputMetadata . Select ( kv => new OnnxVariableInfo ( kv . Key , kv . Value . Dimensions . ToList ( ) , kv . Value . ElementType ) ) ;
169169 }
170170
171171 /// <summary>
172172 /// Returns output metadata of the ONNX model.
173173 /// </summary>
174174 /// <returns></returns>
175- private IEnumerable < OnnxNodeInfo > GetOutputsInfo ( )
175+ private IEnumerable < OnnxVariableInfo > GetOutputsInfo ( )
176176 {
177- return _session . OutputMetadata . Select ( kv => new OnnxNodeInfo ( kv . Key , kv . Value . Dimensions . ToList ( ) , kv . Value . ElementType ) ) ;
177+ return _session . OutputMetadata . Select ( kv => new OnnxVariableInfo ( kv . Key , kv . Value . Dimensions . ToList ( ) , kv . Value . ElementType ) ) ;
178178 }
179179 }
180180
0 commit comments