Add Rescale Attribute changes

Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I8ac71800d922526aad0a7c351ad1943481208cc2

1 files changed, 4 insertions, 1 deletions

diff --git a/src/TosaDeserialize.cpp b/src/TosaDeserialize.cpp
index 0fb02eb..8799028 100644
--- a/src/TosaDeserialize.cpp
+++ b/src/TosaDeserialize.cpp
@@ -1236,9 +1236,12 @@ std::vector<mlir::Value> TosaMlirOperatorBuilder::build<Op_RESCALE>(
   auto double_round = op_builder->getBoolAttr(attr->double_round());
   auto per_channel = op_builder->getBoolAttr(attr->per_channel());
 
+  auto input_unsigned = op_builder->getBoolAttr(attr->input_unsigned());
+  auto output_unsigned = op_builder->getBoolAttr(attr->output_unsigned());
+
   mlir::Operation *mlir_op = op_builder->create<mlir::tosa::RescaleOp>(
       loc, output_type, input_val, input_zp, output_zp, multiplier, shift,
-      scale32, double_round, per_channel);
+      scale32, double_round, per_channel, input_unsigned, output_unsigned);
   block->push_back(mlir_op);
   return std::vector<mlir::Value>({mlir_op->getResult(0)});
 }