Pārlūkot izejas kodu

Update megengine.js

Lutz Roeder 2 gadi atpakaļ
vecāks
revīzija
3fd97f56c9
1 mainītis faili ar 2 papildinājumiem un 1 dzēšanām
  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' ],