diff options
Diffstat (limited to 'src/TosaDeserialize.cpp')
| -rw-r--r-- | src/TosaDeserialize.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/TosaDeserialize.cpp b/src/TosaDeserialize.cpp index a14e69f..f1b7d98 100644 --- a/src/TosaDeserialize.cpp +++ b/src/TosaDeserialize.cpp @@ -1159,7 +1159,7 @@ TosaMlirOperatorBuilder::build<Op_MUL>(TosaSerializationOperator *op) const { Attribute_MulAttribute); // double check attribute type TosaMulAttribute *attr = static_cast<TosaMulAttribute *>(op->GetAttribute()); - auto shift = op_builder->getI32IntegerAttr(attr->shift()); + auto shift = op_builder->getI8IntegerAttr(attr->shift()); mlir::Operation *mlir_op = op_builder->create<mlir::tosa::MulOp>( loc, output_type, input0_val, input1_val, shift); |