Explorar o código

Update paddle.js

Lutz Roeder %!s(int64=3) %!d(string=hai) anos
pai
achega
0bc02b53f3
Modificáronse 1 ficheiros con 5 adicións e 0 borrados
  1. 5 0
      source/paddle.js

+ 5 - 0
source/paddle.js

@@ -793,7 +793,12 @@ paddle.Utility = class {
         const size = tensorDesc.dims.reduce((a, b) => a * b.toNumber(), 1);
         let itemsize = 0;
         switch (tensorDesc.data_type) {
+            case paddle.DataType.FP16: itemsize = 2; break;
             case paddle.DataType.FP32: itemsize = 4; break;
+            case paddle.DataType.FP64: itemsize = 8; break;
+            case paddle.DataType.INT16: itemsize = 2; break;
+            case paddle.DataType.INT32: itemsize = 4; break;
+            case paddle.DataType.INT64: itemsize = 8; break;
             default: throw new paddle.Error("Invalid inference params data type '" + tensorDesc.data_type + "'.");
         }
         const type = paddle.Utility.createTensorType(tensorDesc.data_type, tensorDesc.dims);