Quellcode durchsuchen

Update executorch.js

Lutz Roeder vor 1 Monat
Ursprung
Commit
4461b5259d
2 geänderte Dateien mit 2 neuen und 0 gelöschten Zeilen
  1. 1 0
      source/executorch-schema.js
  2. 1 0
      source/executorch.js

+ 1 - 0
source/executorch-schema.js

@@ -18,6 +18,7 @@ executorch_flatbuffer.ScalarType = {
     QINT8: 12,
     QINT8: 12,
     QUINT8: 13,
     QUINT8: 13,
     QINT32: 14,
     QINT32: 14,
+    BFLOAT16: 15,
     QUINT4X2: 16,
     QUINT4X2: 16,
     QUINT2X4: 17,
     QUINT2X4: 17,
     BITS16: 22,
     BITS16: 22,

+ 1 - 0
source/executorch.js

@@ -246,6 +246,7 @@ executorch.TensorType = class {
             case ScalarType.HALF: this.dataType = 'float16'; break;
             case ScalarType.HALF: this.dataType = 'float16'; break;
             case ScalarType.FLOAT: this.dataType = 'float32'; break;
             case ScalarType.FLOAT: this.dataType = 'float32'; break;
             case ScalarType.DOUBLE: this.dataType = 'float64'; break;
             case ScalarType.DOUBLE: this.dataType = 'float64'; break;
+            case ScalarType.BFLOAT16: this.dataType = 'bfloat16'; break;
             case 8: this.dataType = 'complex<float16>'; break;
             case 8: this.dataType = 'complex<float16>'; break;
             case 9: this.dataType = 'complex<float32>'; break;
             case 9: this.dataType = 'complex<float32>'; break;
             case 10: this.dataType = 'complex<float64>'; break;
             case 10: this.dataType = 'complex<float64>'; break;