Home Explore Help
Sign In
SunnyMirror
/
netron
mirror of https://github.com/lutzroeder/netron.git
2
0
Fork 0
Files Issues 0 Wiki
Browse Source

Update paddle.js

Lutz Roeder 3 years ago
parent
0e9c4da0c9
commit
670a76c4fc
1 changed files with 2 additions and 0 deletions
Split View Show Diff Stats
  1. 2 0
      source/paddle.js

+ 2 - 0
source/paddle.js
View File 

@@ -806,9 +806,11 @@ paddle.Utility = class {
 
             case paddle.DataType.FP16: itemsize = 2; break;
 
             case paddle.DataType.FP32: itemsize = 4; break;
 
             case paddle.DataType.FP64: itemsize = 8; break;
 
+            case paddle.DataType.INT8: itemsize = 1; break;
 
             case paddle.DataType.INT16: itemsize = 2; break;
 
             case paddle.DataType.INT32: itemsize = 4; break;
 
             case paddle.DataType.INT64: itemsize = 8; break;
 
+            case paddle.DataType.UINT8: itemsize = 1; break;
 
             default: throw new paddle.Error("Invalid inference params data type '" + tensorDesc.data_type + "'.");
 
         }
 
         const type = paddle.Utility.createTensorType(tensorDesc.data_type, tensorDesc.dims);