[tflchef/reverse] Enable INT32 type (#2393)
author박세희/동작제어Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Fri, 23 Nov 2018 08:01:07 +0000 (17:01 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Fri, 23 Nov 2018 08:01:07 +0000 (17:01 +0900)
This will enable tflchef-reverse to handle INT32 type
- Needed to handle Reshape operator from MobileNet

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
contrib/tflchef/tflite/src/Convert.cpp

index 37a2cfb..da8e453 100644 (file)
@@ -25,9 +25,10 @@ tflchef::TensorType as_tflchef_type(const tflite::TensorType type)
   {
   case tflite::TensorType_FLOAT32:
     return tflchef::FLOAT32;
+  case tflite::TensorType_INT32:
+    return tflchef::INT32;
   // TODO handle other types
   // TensorType_FLOAT16
-  // TensorType_INT32:
   // TensorType_UINT8
   // TensorType_INT64
   // TensorType_STRING