Skip to content

Commit 44b2a81

Browse files
committed
revert name changes
1 parent 17796fa commit 44b2a81

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

src/Microsoft.ML.OnnxTransformer/OnnxUtils.cs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -29,10 +29,10 @@ internal sealed class OnnxModel
2929
/// </summary>
3030
public sealed class OnnxModelInfo
3131
{
32-
public readonly OnnxVariableInfo[] InputsInfo;
33-
public readonly OnnxVariableInfo[] OutputsInfo;
32+
public readonly OnnxNodeInfo[] InputsInfo;
33+
public readonly OnnxNodeInfo[] OutputsInfo;
3434

35-
public OnnxModelInfo(IEnumerable<OnnxVariableInfo> inputsInfo, IEnumerable<OnnxVariableInfo> outputsInfo)
35+
public OnnxModelInfo(IEnumerable<OnnxNodeInfo> inputsInfo, IEnumerable<OnnxNodeInfo> outputsInfo)
3636
{
3737
InputsInfo = inputsInfo.ToArray();
3838
OutputsInfo = outputsInfo.ToArray();
@@ -43,7 +43,7 @@ public OnnxModelInfo(IEnumerable<OnnxVariableInfo> inputsInfo, IEnumerable<OnnxV
4343
/// OnnxNodeInfo contains all the information for a given node (e.g. inputs/outputs)
4444
/// of an Onnx model.
4545
/// </summary>
46-
public class OnnxVariableInfo
46+
public class OnnxNodeInfo
4747
{
4848
/// <summary>
4949
/// The Name of the node
@@ -58,7 +58,7 @@ public class OnnxVariableInfo
5858
/// </summary>
5959
public readonly System.Type Type;
6060

61-
public OnnxVariableInfo(string name, OnnxShape shape, System.Type type)
61+
public OnnxNodeInfo(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<OnnxVariableInfo> GetInputsInfo()
166+
private IEnumerable<OnnxNodeInfo> GetInputsInfo()
167167
{
168-
return _session.InputMetadata.Select(kv => new OnnxVariableInfo(kv.Key, kv.Value.Dimensions.ToList(), kv.Value.ElementType));
168+
return _session.InputMetadata.Select(kv => new OnnxNodeInfo(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<OnnxVariableInfo> GetOutputsInfo()
175+
private IEnumerable<OnnxNodeInfo> GetOutputsInfo()
176176
{
177-
return _session.OutputMetadata.Select(kv => new OnnxVariableInfo(kv.Key, kv.Value.Dimensions.ToList(), kv.Value.ElementType));
177+
return _session.OutputMetadata.Select(kv => new OnnxNodeInfo(kv.Key, kv.Value.Dimensions.ToList(), kv.Value.ElementType));
178178
}
179179
}
180180

0 commit comments

Comments
 (0)