Add int32 to casts types.

Related to issue kraiskil#26
GorrG · May 19, 2024 · 44fc920 · 44fc920
1 parent f861b9a
commit 44fc920
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/nodes/cast.cc b/src/nodes/cast.cc
@@ -25,11 +25,12 @@ void Cast::resolve(void)
 
 	switch(to)
 	{
+		case onnx::TensorProto_DataType_INT32:
+			output_type = "int32_t"; break;
 		case onnx::TensorProto_DataType_FLOAT:
-			output_type = "float";
+			output_type = "float";   break;
 		case onnx::TensorProto_DataType_DOUBLE:
-			output_type = "double";
-			break;
+			output_type = "double";  break;
 		default:
 			ERROR("Unimplemented casting to requested type");
 	}