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(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(initializer);