Browse Source

Update onnx-proto.js

Lutz Roeder 4 months ago
parent
commit
7297e75268
2 changed files with 13 additions and 4 deletions
  1. 5 2
      source/onnx-proto.js
  2. 8 2
      source/onnx.js

+ 5 - 2
source/onnx-proto.js

@@ -14,7 +14,8 @@ onnx.Version = {
     "IR_VERSION_2023_5_5": 9,
     "IR_VERSION_2024_3_25": 10,
     "IR_VERSION_2025_05_12": 11,
-    "IR_VERSION": 12
+    "IR_VERSION_2025_08_26": 12,
+    "IR_VERSION": 13
 };
 
 onnx.AttributeProto = class AttributeProto {
@@ -1597,7 +1598,9 @@ onnx.TensorProto.DataType = {
     "UINT4": 21,
     "INT4": 22,
     "FLOAT4E2M1": 23,
-    "FLOAT8E8M0": 24
+    "FLOAT8E8M0": 24,
+    "UINT2": 25,
+    "INT2": 26
 };
 
 onnx.TensorProto.Segment = class Segment {

+ 8 - 2
source/onnx.js

@@ -532,6 +532,7 @@ onnx.Tensor = class {
                                 this._encoding = '<';
                             }
                             break;
+                        case onnx.DataType.FLOAT8E8M0:
                         case onnx.DataType.FLOAT4E2M1:
                         case onnx.DataType.FLOAT8E4M3FN:
                         case onnx.DataType.FLOAT8E4M3FNUZ:
@@ -542,8 +543,10 @@ onnx.Tensor = class {
                                 this._encoding = '<';
                             }
                             break;
-                        case onnx.DataType.UINT4:
+                        case onnx.DataType.INT2:
                         case onnx.DataType.INT4:
+                        case onnx.DataType.UINT2:
+                        case onnx.DataType.UINT4:
                             if (tensor.int32_data && tensor.int32_data.length > 0) {
                                 this._data = new Uint8Array(Array.from(tensor.int32_data));
                                 this._encoding = '<';
@@ -1273,7 +1276,10 @@ onnx.DataType = {
     FLOAT8E5M2FNUZ: 20,
     UINT4: 21,
     INT4: 22,
-    FLOAT4E2M1: 23
+    FLOAT4E2M1: 23,
+    FLOAT8E8M0: 24,
+    UINT2: 25,
+    INT2: 26
 };
 
 onnx.AttributeType = {