From cd2c5f5d944f7fba47dbf5108486932cf6b535c0 Mon Sep 17 00:00:00 2001 From: David Monahan Date: Fri, 9 Oct 2020 12:30:04 +0100 Subject: IVGCVSW-5291 Fix for Yolov3 producing 0s on Neon Signed-off-by: David Monahan Change-Id: I9331d590e71fc479979c11be9dc750d6435f12bc --- src/armnn/layers/ConcatLayer.cpp | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/armnn/layers/ConcatLayer.cpp b/src/armnn/layers/ConcatLayer.cpp index 0118426954..affbe7148e 100644 --- a/src/armnn/layers/ConcatLayer.cpp +++ b/src/armnn/layers/ConcatLayer.cpp @@ -89,6 +89,14 @@ void ConcatLayer::CreateTensors(const TensorHandleFactoryRegistry& registry, } } + // Splitter layer outputs are subtensors on the inputs whereas concat inputs are subtensors on + // the output. If the parent is a Splitter layer we cannot use subtensors. + if ((PolymorphicDowncast(&(slot->GetOwningLayer())))->GetType() == LayerType::Splitter + && (PolymorphicDowncast(currentLayer))->GetType() == LayerType::Concat) + { + canUseSubTensorOnXorY = false; + } + if (!canUseSubTensorOnXorY) { break; -- cgit v1.2.1