From cf4b0bba07c00b0f0921f994d9b81baef0f9f530 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=A2=85=ED=98=84/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Thu, 6 Sep 2018 19:14:27 +0900 Subject: [PATCH] [enco] Support INT32 tensor type (#1389) This commit extends ANN CodeGen to properly declare a tensor operand of INT32 type. Signed-off-by: Jonghyun Park --- contrib/enco/core/src/CppGen/Subnet.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/contrib/enco/core/src/CppGen/Subnet.cpp b/contrib/enco/core/src/CppGen/Subnet.cpp index 084a102..f927583 100644 --- a/contrib/enco/core/src/CppGen/Subnet.cpp +++ b/contrib/enco/core/src/CppGen/Subnet.cpp @@ -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: -- 2.7.4