diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/TosaDeserialize.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/TosaDeserialize.cpp b/src/TosaDeserialize.cpp index db52d67..335a997 100644 --- a/src/TosaDeserialize.cpp +++ b/src/TosaDeserialize.cpp @@ -338,7 +338,7 @@ TosaMlirOperatorBuilder::BuildReductionOp(TosaSerializationOperator *op) const { TosaAxisAttribute *attr = static_cast<TosaAxisAttribute *>(op->GetAttribute()); - auto axis = op_builder->getI64IntegerAttr(attr->axis()); + auto axis = op_builder->getI32IntegerAttr(attr->axis()); mlir::Operation *mlir_op = op_builder->create<MLIR_OP>(loc, output_type, input_val, axis); @@ -711,7 +711,7 @@ TosaMlirOperatorBuilder::build<Op_CONCAT>(TosaSerializationOperator *op) const { Attribute_AxisAttribute); // double check attribute type TosaAxisAttribute *attr = static_cast<TosaAxisAttribute *>(op->GetAttribute()); - auto axis = op_builder->getI64IntegerAttr(attr->axis()); + auto axis = op_builder->getI32IntegerAttr(attr->axis()); mlir::Operation *mlir_op = op_builder->create<mlir::tosa::ConcatOp>( loc, output_type, input_values, axis); |