From f59b16f42ef68bde877b70816ffb953d64c8baa3 Mon Sep 17 00:00:00 2001 From: Isabella Gottardi Date: Thu, 25 Jul 2019 12:03:39 +0100 Subject: COMPMID-2336: Set the correct valid region to the concatenation layer output Change-Id: I1ff84054758a9a189f33d3180b5d81b3615963c7 Signed-off-by: Isabella Gottardi Reviewed-on: https://review.mlplatform.org/c/1620 Reviewed-by: Gian Marco Iodice Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- src/core/NEON/kernels/NEWidthConcatenateLayerKernel.cpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/core/NEON/kernels/NEWidthConcatenateLayerKernel.cpp') diff --git a/src/core/NEON/kernels/NEWidthConcatenateLayerKernel.cpp b/src/core/NEON/kernels/NEWidthConcatenateLayerKernel.cpp index 7b1ad9c2e8..cafa20a1bd 100644 --- a/src/core/NEON/kernels/NEWidthConcatenateLayerKernel.cpp +++ b/src/core/NEON/kernels/NEWidthConcatenateLayerKernel.cpp @@ -93,6 +93,9 @@ void NEWidthConcatenateLayerKernel::configure(const ITensor *input, unsigned int ARM_COMPUTE_ERROR_THROW_ON(std::get<0>(win_config)); INEKernel::configure(std::get<1>(win_config)); + + // Set output valid region + output->info()->set_valid_region(ValidRegion(Coordinates(), output->info()->tensor_shape())); } Status NEWidthConcatenateLayerKernel::validate(const ITensorInfo *input, unsigned int width_offset, const ITensorInfo *output) -- cgit v1.2.1