commit e7a5b0e133b977ef80d31e86f4eb6e94eae5ba17
author Pablo Marquez Tello <pablo.tello@arm.com> Wed Nov 10 15:41:57 2021 +0000
committer Sheri Zhang <sheri.zhang@arm.com> Mon Nov 15 13:09:15 2021 +0000
tree 5f2380ab1c2ad92fd49123bac06e7f68b446ce5c
parent 3177861f8145640761e8184017f4836255a7c293

Initialise quantization info in NEPadLayer

* Resolves COMPMID-4884

Signed-off-by: Pablo Marquez Tello <pablo.tello@arm.com>
Change-Id: Id28f0df242fe240c70f22e3ad55e4729ab1e40fe
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6641
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Sheri Zhang <sheri.zhang@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>

src/runtime/NEON/functions/NEPadLayer.cpp

diff --git a/src/runtime/NEON/functions/NEPadLayer.cpp b/src/runtime/NEON/functions/NEPadLayer.cpp
index e776e67..8bacdd3 100644
--- a/src/runtime/NEON/functions/NEPadLayer.cpp
+++ b/src/runtime/NEON/functions/NEPadLayer.cpp
@@ -153,6 +153,11 @@
             }
             // 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)
             {