2
0
Lutz Roeder 1 жил өмнө
parent
commit
57b73b4d01

+ 4 - 0
source/espresso-metadata.json

@@ -70,5 +70,9 @@
   {
     "name": "squeeze",
     "category": "Transform"
+  },
+  {
+    "name": "rnn_arch",
+    "category": "layer"
   }
 ]

+ 5 - 0
source/espresso.js

@@ -404,6 +404,11 @@ espresso.Reader = class {
             delete data.blob_weights;
             return;
         }
+        if (data.blob_weights_f16 !== undefined) {
+            obj.inputs.push(this._initializer('weights', data.blob_weights_f16, 'float16', dimensions));
+            delete data.blob_weights_f16;
+            return;
+        }
         const keys = ['wBeta', 'wGamma', 'W_S8', 'W_int8', 'W_t_int8'];
         for (const key of keys) {
             if (data.weights && data.weights[key] !== undefined) {