Skip to content

Commit 9018d9c

Browse files
committed
fix: Fix data type to string
1 parent e099d23 commit 9018d9c

File tree

1 file changed

+14
-14
lines changed

1 file changed

+14
-14
lines changed

cpp/TensorHelpers.cpp

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -23,24 +23,26 @@ std::string dataTypeToString(TfLiteType dataType) {
2323
switch (dataType) {
2424
case kTfLiteFloat32:
2525
return "float32";
26+
case kTfLiteFloat64:
27+
return "float64";
28+
case kTfLiteInt4:
29+
return "int4";
30+
case kTfLiteInt8:
31+
return "int8";
32+
case kTfLiteInt16:
33+
return "int16";
2634
case kTfLiteInt32:
2735
return "int32";
28-
case kTfLiteUInt8:
29-
return "int8";
3036
case kTfLiteInt64:
3137
return "int64";
32-
case kTfLiteInt16:
33-
return "int16";
34-
case kTfLiteInt8:
35-
return "int8";
36-
case kTfLiteFloat64:
37-
return "float64";
38+
case kTfLiteUInt8:
39+
return "uint8";
40+
case kTfLiteUInt16:
41+
return "uint16";
42+
case kTfLiteUInt32:
43+
return "uint32";
3844
case kTfLiteUInt64:
3945
return "uint64";
40-
case kTfLiteUInt32:
41-
return "int32";
42-
case kTfLiteUInt16:
43-
return "int16";
4446
case kTfLiteNoType:
4547
return "none";
4648
case kTfLiteString:
@@ -55,8 +57,6 @@ std::string dataTypeToString(TfLiteType dataType) {
5557
return "resource";
5658
case kTfLiteVariant:
5759
return "variant";
58-
case kTfLiteInt4:
59-
return "int4";
6060
default:
6161
[[unlikely]];
6262
return "invalid";

0 commit comments

Comments
 (0)