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;