From ce48d1dda2d6b32477de129a05ca3adee0f87690 Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Wed, 24 Apr 2024 13:30:58 +0100 Subject: IVGCVSW-8206 CreateRescaleTosaOperator() modified Signed-off-by: Teresa Charlin Signed-off-by: Cathal Corbett Change-Id: I520a9d6cc95faeefaf82947a0547292e869aa523 --- src/backends/tosaCommon/operatorMappings/Conv2dOperator.cpp | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) (limited to 'src/backends/tosaCommon/operatorMappings/Conv2dOperator.cpp') diff --git a/src/backends/tosaCommon/operatorMappings/Conv2dOperator.cpp b/src/backends/tosaCommon/operatorMappings/Conv2dOperator.cpp index 1c8682b1ab..96701d4384 100644 --- a/src/backends/tosaCommon/operatorMappings/Conv2dOperator.cpp +++ b/src/backends/tosaCommon/operatorMappings/Conv2dOperator.cpp @@ -140,7 +140,6 @@ TosaSerializationBasicBlock* ConvertConv2dToTosaOperator(const Layer* layer, {convOutStr}); operators.push_back(conv2d_op); - if (isInputInt8) { int32_t output_zp = outputs[0]->GetQuantizationOffset(); @@ -149,11 +148,8 @@ TosaSerializationBasicBlock* ConvertConv2dToTosaOperator(const Layer* layer, const std::vector& weight_scales = inputs[1]->GetQuantizationScales(); TosaSerializationOperator* rescaleOp = nullptr; - TosaSerializationTensor* rescaleTensor = nullptr; CreateRescaleTosaOperatorPerChannel(outputConv2dName, outputName, - DType_INT8, - outputShape0, 0, output_zp, true, @@ -161,11 +157,13 @@ TosaSerializationBasicBlock* ConvertConv2dToTosaOperator(const Layer* layer, input_scale, output_scale, weight_scales, - &rescaleOp, - &rescaleTensor); + &rescaleOp); operators.push_back(rescaleOp); - tensors.push_back(rescaleTensor); + tensors.push_back(new TosaSerializationTensor(outputName, + outputShape0, + DType_INT8, {})); } + // operatorInputNames/operatorOutputNames ends up being the same as // blockInputNames/blockOutputNames for one-to-one ArmNN to TOSA mappings return new TosaSerializationBasicBlock(blockName, // name -- cgit v1.2.1