diff options
Diffstat (limited to 'src/armnn/layers')
-rw-r--r-- | src/armnn/layers/ConcatLayer.cpp | 8 |
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; |