diff options
author | David Monahan <david.monahan@arm.com> | 2020-10-09 12:30:04 +0100 |
---|---|---|
committer | David Monahan <david.monahan@arm.com> | 2020-10-09 15:06:46 +0000 |
commit | cd2c5f5d944f7fba47dbf5108486932cf6b535c0 (patch) | |
tree | e680dddbdd795ba87d40afc933bdc4cad8fa1354 /src/armnn/layers/ConcatLayer.cpp | |
parent | d63e86c016c7c8049874da3dfc0576d25ddc65fb (diff) | |
download | armnn-cd2c5f5d944f7fba47dbf5108486932cf6b535c0.tar.gz |
IVGCVSW-5291 Fix for Yolov3 producing 0s on Neon
Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I9331d590e71fc479979c11be9dc750d6435f12bc
Diffstat (limited to 'src/armnn/layers/ConcatLayer.cpp')
-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; |