aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/MergerLayer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/layers/MergerLayer.cpp')
-rw-r--r--src/armnn/layers/MergerLayer.cpp5
1 files changed, 4 insertions, 1 deletions
diff --git a/src/armnn/layers/MergerLayer.cpp b/src/armnn/layers/MergerLayer.cpp
index e80661a493..545c7e0e0e 100644
--- a/src/armnn/layers/MergerLayer.cpp
+++ b/src/armnn/layers/MergerLayer.cpp
@@ -40,7 +40,10 @@ void MergerLayer::CreateTensorHandles(Graph& graph, const IWorkloadFactory& fact
//just needs to make sure that the outputs of the prev layer
//are made subtensors of the output of the merger layer.
m_OutputHandlers[0].CreateTensorHandles(factory);
- if (factory.SupportsSubTensors())
+
+ signed long innerAxis = m_Param.GetNumDimensions() - m_Param.GetConcatAxis();
+
+ if (factory.SupportsSubTensors() && innerAxis != 1)
{
std::queue<MergerLayer*> m_MergerLayers;