[enco] Support INT32 tensor type (#1389)
author박종현/동작제어Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Thu, 6 Sep 2018 10:14:27 +0000 (19:14 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Thu, 6 Sep 2018 10:14:27 +0000 (19:14 +0900)
This commit extends ANN CodeGen to properly declare a tensor operand of
INT32 type.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
contrib/enco/core/src/CppGen/Subnet.cpp

index 084a102..f927583 100644 (file)
@@ -86,6 +86,8 @@ const char *tensor_operand_code(const ann::DType &dtype)
 {
   switch (dtype)
   {
+  case ann::DType::S32:
+    return "ANEURALNETWORKS_TENSOR_INT32";
   case ann::DType::F32:
     return "ANEURALNETWORKS_TENSOR_FLOAT32";
   default: