From e7a5b0e133b977ef80d31e86f4eb6e94eae5ba17 Mon Sep 17 00:00:00 2001 From: Pablo Marquez Tello Date: Wed, 10 Nov 2021 15:41:57 +0000 Subject: Initialise quantization info in NEPadLayer * Resolves COMPMID-4884 Signed-off-by: Pablo Marquez Tello Change-Id: Id28f0df242fe240c70f22e3ad55e4729ab1e40fe Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6641 Tested-by: Arm Jenkins Reviewed-by: Sheri Zhang Comments-Addressed: Arm Jenkins --- src/runtime/NEON/functions/NEPadLayer.cpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/runtime/NEON/functions') diff --git a/src/runtime/NEON/functions/NEPadLayer.cpp b/src/runtime/NEON/functions/NEPadLayer.cpp index e776e673c2..8bacdd3002 100644 --- a/src/runtime/NEON/functions/NEPadLayer.cpp +++ b/src/runtime/NEON/functions/NEPadLayer.cpp @@ -153,6 +153,11 @@ void NEPadLayer::configure_reflect_symmetric_mode(ITensor *input, ITensor *outpu } // Concatenate the padding before and after with the input. ITensor *out = (i == _num_dimensions - 1) ? output : &_concat_results[i]; + out->info()->set_quantization_info(output->info()->quantization_info()); + for(auto &v : concat_vector) + { + v->info()->set_quantization_info(input->info()->quantization_info()); + } _concat_functions[i].configure(concat_vector, out, i); if(i != _num_dimensions - 1) { -- cgit v1.2.1