From e683d9693e16ac9bee21a614e0d651b741c311eb Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Principal=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Fri, 23 Nov 2018 17:01:07 +0900 Subject: [PATCH] [tflchef/reverse] Enable INT32 type (#2393) This will enable tflchef-reverse to handle INT32 type - Needed to handle Reshape operator from MobileNet Signed-off-by: SaeHie Park --- contrib/tflchef/tflite/src/Convert.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/contrib/tflchef/tflite/src/Convert.cpp b/contrib/tflchef/tflite/src/Convert.cpp index 37a2cfb..da8e453 100644 --- a/contrib/tflchef/tflite/src/Convert.cpp +++ b/contrib/tflchef/tflite/src/Convert.cpp @@ -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 -- 2.7.4