فهرست منبع

Update mslite-schema.js

Lutz Roeder 2 سال پیش
والد
کامیت
c713b46fa8
2فایلهای تغییر یافته به همراه12 افزوده شده و 2 حذف شده
  1. 4 1
      source/mslite-metadata.json
  2. 8 1
      source/mslite-schema.js

+ 4 - 1
source/mslite-metadata.json

@@ -1023,7 +1023,10 @@
     "name": "DynamicQuant",
     "attributes": [
       { "name": "symmetric", "type": "boolean", "default": false },
-      { "name": "dst_type", "type": "int64", "default": 32 }
+      { "name": "dst_type", "type": "int64", "default": 32 },
+      { "name": "activation_channel", "type": "boolean", "default": false },
+      { "name": "prefer_axis", "type": "int64", "default": 0 },
+      { "name": "transpose", "type": "boolean", "default": false }
     ]
   },
   {

+ 8 - 1
source/mslite-schema.js

@@ -43,7 +43,8 @@ $root.mindspore.schema.Format = {
     NUM_OF_FORMAT: 14,
     NCDHW: 15,
     NWC: 16,
-    NCW: 17
+    NCW: 17,
+    NC8HW8: 18
 };
 
 $root.mindspore.schema.ActivationType = {
@@ -4052,6 +4053,9 @@ $root.mindspore.schema.DynamicQuant = class DynamicQuant {
         const $ = new $root.mindspore.schema.DynamicQuant();
         $.symmetric = reader.bool_(position, 4, false);
         $.dst_type = reader.int64_(position, 6, 32);
+        $.activation_channel = reader.bool_(position, 8, false);
+        $.prefer_axis = reader.int64_(position, 10, 0);
+        $.transpose = reader.bool_(position, 12, false);
         return $;
     }
 
@@ -4059,6 +4063,9 @@ $root.mindspore.schema.DynamicQuant = class DynamicQuant {
         const $ = new $root.mindspore.schema.DynamicQuant();
         $.symmetric = reader.value(json.symmetric, false);
         $.dst_type = reader.value(json.dst_type, 32);
+        $.activation_channel = reader.value(json.activation_channel, false);
+        $.prefer_axis = reader.value(json.prefer_axis, 0);
+        $.transpose = reader.value(json.transpose, false);
         return $;
     }
 };