Update paddle-proto.js

source/paddle-proto.js

@@ -404,6 +404,7 @@ $root.paddle.framework.proto.VarType.Type = {
     "SIZE_T": 19,
     "UINT8": 20,
     "INT8": 21,
+    "BF16": 22,
     "LOD_TENSOR": 7,
     "SELECTED_ROWS": 8,
     "FEED_MINIBATCH": 9,

source/paddle.js

@@ -543,18 +543,36 @@ paddle.TensorType = class {
 
     constructor(desc) {
         switch (desc.data_type) {
+            case paddle.proto.VarType.Type.BOOL:
+                this._dataType = 'boolean';
+                break;
+            case paddle.proto.VarType.Type.INT8:
+                this._dataType = 'int8';
+                break;
+            case paddle.proto.VarType.Type.INT16:
+                this._dataType = 'int16';
+                break;
             case paddle.proto.VarType.Type.INT32:
                 this._dataType = 'int32';
                 break;
             case paddle.proto.VarType.Type.INT64:
                 this._dataType = 'int64';
                 break;
+            case paddle.proto.VarType.Type.BP16:
+                this._dataType = 'bfloat16';
+                break;
+            case paddle.proto.VarType.Type.FP16:
+                this._dataType = 'float16';
+                break;
             case paddle.proto.VarType.Type.FP32:
                 this._dataType = 'float32';
                 break;
             case paddle.proto.VarType.Type.FP64:
                 this._dataType = 'float64';
                 break;
+            case paddle.proto.VarType.Type.UINT8:
+                this._dataType = 'uint8';
+                break;
             default:
                 this._dataType = '?';
                 break;