IVGCVSW-5291 Fix for Yolov3 producing 0s on Neon

Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I9331d590e71fc479979c11be9dc750d6435f12bc

1 files changed, 8 insertions, 0 deletions

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<const Layer*>(&(slot->GetOwningLayer())))->GetType() == LayerType::Splitter
+                        && (PolymorphicDowncast<const Layer*>(currentLayer))->GetType() == LayerType::Concat)
+                    {
+                        canUseSubTensorOnXorY = false;
+                    }
+
                     if (!canUseSubTensorOnXorY)
                     {
                         break;