From c7b5eecc7f60df87ea094e6823683d665d88bc60 Mon Sep 17 00:00:00 2001
From: Lsz <lester.szeto@gmail.com>
Date: Thu, 13 Aug 2020 06:16:27 -0700
Subject: [PATCH] Added builder case for int8 tensors in the dialect (#263)

---
 src/Builder/FrontendDialectHelper.cpp | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/Builder/FrontendDialectHelper.cpp b/src/Builder/FrontendDialectHelper.cpp
index 4efad0a..3c780c7 100644
--- a/src/Builder/FrontendDialectHelper.cpp
+++ b/src/Builder/FrontendDialectHelper.cpp
@@ -166,6 +166,15 @@ mlir::DenseElementsAttr onnxTensorProtoToDenseElmAttr(
         tensorType, llvm::makeArrayRef(arrayAttrInitializer));
     break;
   }
+  case (onnx::TensorProto::INT8): {
+    const auto &arrayAttrInitializer =
+        CreateArrayAttribute<int32_t>(initializer);
+    auto elmType = builder.getIntegerType(8);
+    auto tensorType = mlir::RankedTensorType::get(tensorDims, elmType);
+    denseElmAttr = mlir::DenseElementsAttr::get(
+        tensorType, llvm::makeArrayRef(arrayAttrInitializer));
+    break;
+  }
   case (onnx::TensorProto::INT32): {
     const auto &arrayAttrInitializer =
         CreateArrayAttribute<int32_t>(initializer);