소스 검색

Update rknn.js

Lutz Roeder 1 개월 전
부모
커밋
29ab06101f
1개의 변경된 파일3개의 추가작업 그리고 1개의 파일을 삭제
  1. 3 1
      source/rknn.js

+ 3 - 1
source/rknn.js

@@ -183,7 +183,7 @@ rknn.Graph = class {
             }
             case 'flatbuffers': {
                 const graph = obj;
-                const dataTypes = ['undefined', 'float32', 'uint8', 'int8', 'uint16', 'int16', 'int32', 'int64', 'string', 'boolean', 'float16', 'float64', 'uint32', 'uint64', 'complex<float32>', 'complex<float64>', 'bfloat16'];
+                const dataTypes = ['?', 'float32', 'uint8', 'int8', 'uint16', 'int16', 'int32', 'int64', 'string', 'boolean', 'float16', 'float64', 'uint32', 'uint64', 'complex<float32>', 'complex<float64>', 'bfloat16'];
                 const args = graph.tensors.map((tensor) => {
                     const shape = new rknn.TensorShape(Array.from(tensor.shape));
                     const dataType = tensor.data_type < dataTypes.length ? dataTypes[tensor.data_type] : '?';
@@ -359,6 +359,8 @@ rknn.Tensor = class {
             case 'float64': itemsize = 8; break;
             case 'boolean': itemsize = 1; break;
             case 'vdata': itemsize = 1; break;
+            case 'string': itemsize = 1; break;
+            case '?': itemsize = 0; break;
             default: throw new rknn.Error(`Unsupported tensor data type '${this.type.dataType}'.`);
         }
         if (weights) {