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 ab60a26..d55585e 100644 --- a/src/TosaDeserialize.cpp +++ b/src/TosaDeserialize.cpp @@ -425,6 +425,13 @@ TosaMlirOperatorBuilder::build<Op_CONST>(TosaSerializationOperator *op) const { 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); |