Skip to content

Commit 8ec5398

Browse files
committed
[TMVA][SOFIE] Also support ConvertTypeToString for 8-bit integers
1 parent 1561f5c commit 8ec5398

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tmva/sofie/src/SOFIE_common.cxx

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,9 @@ std::string ConvertTypeToString(ETensorType type){
5959
case ETensorType::FLOAT : {
6060
return "float";
6161
}
62+
case ETensorType::INT8 : {
63+
return "int8_t";
64+
}
6265
case ETensorType::INT16 : {
6366
return "int16_t";
6467
}
@@ -68,6 +71,9 @@ std::string ConvertTypeToString(ETensorType type){
6871
case ETensorType::INT64 : {
6972
return "int64_t";
7073
}
74+
case ETensorType::UINT8 : {
75+
return "uint8_t";
76+
}
7177
case ETensorType::UINT16 : {
7278
return "uint16_t";
7379
}

0 commit comments

Comments
 (0)