Procházet zdrojové kódy

Update megengine.js

Lutz Roeder před 2 roky
rodič
revize
3fd97f56c9
1 změnil soubory, kde provedl 2 přidání a 1 odebrání
  1. 2 1
      source/megengine.js

+ 2 - 1
source/megengine.js

@@ -561,7 +561,8 @@ megengine.Tensor = class {
 megengine.TensorType = class {
 
     constructor(dataType, shape) {
-        dataType = megengine.Utility.enum(megengine.schema, 'DTypeEnum', dataType).toLowerCase();
+        dataType = megengine.Utility.enum(megengine.schema, 'DTypeEnum', dataType);
+        dataType = typeof dataType === 'string' ? dataType.toLowerCase() : dataType;
         megengine.TensorType._dataTypes = megengine.TensorType._dataTypes || new Map([
             [ 'bool', 'boolean' ],
             [ 'byte', 'uint8' ], [ 'quantizeds4asymm', 'uint8' ], [ 'quantizeds8asymm', 'uint8' ], [ 'uintb4', 'uint8' ],