diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-09-19 16:46:54 +0100 |
---|---|---|
committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-09-19 16:46:54 +0100 |
commit | 0aa080db52c60be959111df92233321eca258429 (patch) | |
tree | 9fe527bc70247086f59bcce6491cd4abd1ea61c3 /src/backends | |
parent | 987398f3c1b93cc63f6c25da8b9a20c68e7e8270 (diff) | |
download | armnn-0aa080db52c60be959111df92233321eca258429.tar.gz |
BugFix: ElementwiseBinary missing from NeonBackend activation fusion optimization
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I384e0613f1dad6a1ebf732dcb17d04a09a369084
Diffstat (limited to 'src/backends')
-rw-r--r-- | src/backends/neon/NeonBackend.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/backends/neon/NeonBackend.cpp b/src/backends/neon/NeonBackend.cpp index a5e9ab9a96..b5719db007 100644 --- a/src/backends/neon/NeonBackend.cpp +++ b/src/backends/neon/NeonBackend.cpp @@ -164,7 +164,8 @@ OptimizationViews NeonBackend::OptimizeSubgraphView(const SubgraphView& subgraph if ((base.GetType() == LayerType::DepthwiseConvolution2d || base.GetType() == LayerType::Convolution2d || base.GetType() == LayerType::BatchNormalization || base.GetType() == LayerType::FullyConnected || base.GetType() == LayerType::Addition || base.GetType() == LayerType::Multiplication - || base.GetType() == LayerType::Subtraction || base.GetType() == LayerType::Division) + || base.GetType() == LayerType::Subtraction || base.GetType() == LayerType::Division + || base.GetType() == LayerType::ElementwiseBinary) && (base.GetAdditionalInformation<ActivationDescriptor>() == nullptr)) { for (auto output = base.BeginOutputSlots(); output != base.EndOutputSlots(); ++output) |