瀏覽代碼

Update onnx-proto.js

Lutz Roeder 2 年之前
父節點
當前提交
1b7d7b85ab
共有 1 個文件被更改,包括 7 次插入0 次删除
  1. 7 0
      source/onnx-proto.js

+ 7 - 0
source/onnx-proto.js

@@ -1502,6 +1502,7 @@ onnx.FunctionProto = class FunctionProto {
         this.attribute_proto = [];
         this.node = [];
         this.opset_import = [];
+        this.value_info = [];
     }
 
     static decode(reader, length) {
@@ -1537,6 +1538,9 @@ onnx.FunctionProto = class FunctionProto {
                 case 10:
                     message.domain = reader.string();
                     break;
+                case 12:
+                    message.value_info.push(onnx.ValueInfoProto.decode(reader, reader.uint32()));
+                    break;
                 default:
                     reader.skipType(tag & 7);
                     break;
@@ -1578,6 +1582,9 @@ onnx.FunctionProto = class FunctionProto {
                 case "domain":
                     message.domain = reader.string();
                     break;
+                case "value_info":
+                    message.value_info.push(onnx.ValueInfoProto.decodeText(reader));
+                    break;
                 default:
                     reader.field(tag, message);
                     break;