Bläddra i källkod

Update rknn.js

Lutz Roeder 3 år sedan
förälder
incheckning
49656aa088
1 ändrade filer med 6 tillägg och 0 borttagningar
  1. 6 0
      source/rknn.js

+ 6 - 0
source/rknn.js

@@ -280,6 +280,7 @@ rknn.Tensor = class {
         switch (this._type.dataType) {
             case 'uint8': size = 1; break;
             case 'int8': size = 1; break;
+            case 'int16': size = 2; break;
             case 'int32': size = 4; break;
             case 'float16': size = 2; break;
             case 'float32': size = 4; break;
@@ -368,6 +369,11 @@ rknn.Tensor = class {
                         context.index += 1;
                         context.count++;
                         break;
+                    case 'int16':
+                        results.push(context.view.getInt16(context.index, true));
+                        context.index += 2;
+                        context.count++;
+                        break;
                     case 'int32':
                         results.push(context.view.getInt32(context.index, true));
                         context.index += 4;