Selaa lähdekoodia

Update coreml-proto.js

Lutz Roeder 7 kuukautta sitten
vanhempi
sitoutus
1525df5cd8
2 muutettua tiedostoa jossa 4 lisäystä ja 0 poistoa
  1. 1 0
      source/coreml-proto.js
  2. 3 0
      source/coreml.js

+ 1 - 0
source/coreml-proto.js

@@ -2299,6 +2299,7 @@ CoreML.Specification.ArrayFeatureType.ArrayDataType = {
     "FLOAT32": 65568,
     "DOUBLE": 65600,
     "INT32": 131104,
+    "INT8": 131080,
     "FLOAT16": 65552
 };
 

+ 3 - 0
source/coreml.js

@@ -1483,6 +1483,9 @@ coreml.Utility = class {
                         case ArrayDataType.INT32:
                             dataType = 'int32';
                             break;
+                        case ArrayDataType.INT8:
+                            dataType = 'int8';
+                            break;
                         default:
                             throw new coreml.Error(`Unsupported array data type '${arrayType.dataType}'.`);
                     }