diff options
author | James Ward <james.ward@arm.com> | 2023-11-14 18:45:33 +0000 |
---|---|---|
committer | James Ward <james.ward@arm.com> | 2023-11-27 10:16:00 +0000 |
commit | fc32f56a067c526238c15de097fe78fdcab95cb5 (patch) | |
tree | c1a4041b7ec9079ab6d1fdce8fcc6395538662e1 /src/TosaSerialize.cpp | |
parent | 546e9990065804f6304a216b42468bf44c8c1036 (diff) | |
download | tosa_mlir_translator-fc32f56a067c526238c15de097fe78fdcab95cb5.tar.gz |
Add Rescale Attribute changes
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I8ac71800d922526aad0a7c351ad1943481208cc2
Diffstat (limited to 'src/TosaSerialize.cpp')
-rw-r--r-- | src/TosaSerialize.cpp | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp index 9807a99..263f51c 100644 --- a/src/TosaSerialize.cpp +++ b/src/TosaSerialize.cpp @@ -1415,6 +1415,11 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::RescaleOp>( op.getAttr("multiplier").dyn_cast<mlir::DenseI32ArrayAttr>().asArrayRef(); auto shift = getDenseI8ArrayAttr<int32_t>(op.getAttr("shift")); + bool input_unsigned = + op.getAttr("input_unsigned").dyn_cast<mlir::BoolAttr>().getValue(); + bool output_unsigned = + op.getAttr("output_unsigned").dyn_cast<mlir::BoolAttr>().getValue(); + auto input = op.getOperand(0); auto input_ty = input.getType().cast<mlir::RankedTensorType>(); auto output = op.getResult(0); @@ -1423,9 +1428,9 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::RescaleOp>( std::string input_name = GetTensorName(input); std::string output_name = GetTensorName(output); - TosaRescaleAttribute attribute( - input_zp, output_zp, multiplier, shift, scale32, double_round, - per_channel, input_ty.isUnsignedInteger(), output_ty.isUnsignedInteger()); + TosaRescaleAttribute attribute(input_zp, output_zp, multiplier, shift, + scale32, double_round, per_channel, + input_unsigned, output_unsigned); TosaSerializationOperator *tyop = new TosaSerializationOperator( Op_RESCALE, Attribute_RescaleAttribute, &attribute, |