diff options
-rw-r--r-- | src/TosaDeserialize.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/TosaDeserialize.cpp b/src/TosaDeserialize.cpp index f6f78fc..da92c19 100644 --- a/src/TosaDeserialize.cpp +++ b/src/TosaDeserialize.cpp @@ -384,6 +384,13 @@ std::vector<mlir::Value> TosaMlirOperatorBuilder::build<Op_CONST>(TosaSerializat mlir::DenseElementsAttr::get(output_type, llvm::ArrayRef(float_data)); break; } + case DType_INT4: { + std::vector<int8_t> int4_data; + TosaSerializationHandler::ConvertU8toI4(data, out_size, int4_data); + value_attr = + mlir::DenseElementsAttr::get(output_type, llvm::ArrayRef(int4_data)); + break; + } case DType_INT8: { std::vector<int8_t> int8_data; TosaSerializationHandler::ConvertU8toI8(data, out_size, int8_data); |