Update mslite.js (#1310)

lutzroeder · Jul 4, 2024 · de0a831 · de0a831
1 parent 463b2c9
commit de0a831
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/source/mslite.js b/source/mslite.js
@@ -260,21 +260,26 @@ mslite.TensorType = class {
  case 27: this.dataType = "refkey"; break;
  case 28: this.dataType = "ref"; break;
  case 30: this.dataType = "boolean"; break;
- case 31: this.dataType = "int"; break;
+ // case 31: this.dataType = "int"; break;
  case 32: this.dataType = "int8"; break;
  case 33: this.dataType = "int16"; break;
  case 34: this.dataType = "int32"; break;
  case 35: this.dataType = "int64"; break;
- case 36: this.dataType = "uint"; break;
+ // case 36: this.dataType = "uint"; break;
  case 37: this.dataType = "uint8"; break;
  case 38: this.dataType = "uint16"; break;
  case 39: this.dataType = "uint32"; break;
  case 40: this.dataType = "uint64"; break;
- case 41: this.dataType = "float"; break;
+ // case 41: this.dataType = "float"; break;
  case 42: this.dataType = "float16"; break;
  case 43: this.dataType = "float32"; break;
  case 44: this.dataType = "float64"; break;
- case 45: this.dataType = "complex64"; break;
+ case 45: this.dataType = "bfloat16"; break;
+ // case 46: this.dataType = "double"; break;
+ // case 47: this.dataType = "complex"; break;
+ case 48: this.dataType = "complex64"; break;
+ case 49: this.dataType = "complex128"; break;
+ case 50: this.dataType = "int4"; break;
  default: throw new mslite.Error(`Unsupported data type '${dataType}'.`);
  }
  this.shape = new mslite.TensorShape(Array.from(dimensions));