aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonBackend.cpp
diff options
context:
space:
mode:
authorMatthew Bentham <matthew.bentham@arm.com>2022-02-08 15:03:07 +0000
committerMatthew Bentham <matthew.bentham@arm.com>2022-02-08 15:25:06 +0000
commit67d6390c3b37c83bad2180ee4dc54893143c7b29 (patch)
tree41b0392841c3e070e60f264ebdf1910be49c9176 /src/backends/neon/NeonBackend.cpp
parentc42a987aa53d0fd842c34dee90abef5a9ff15fa4 (diff)
downloadarmnn-67d6390c3b37c83bad2180ee4dc54893143c7b29.tar.gz
Handle optional biases better in Neon/Cl FullyConnected workloads
Use armnn::Optional for optional bias TensorInfos, similar to how it's already done in Convolution etc. Fixes some test failures found using -fsanitize=undefined Change-Id: I7b887e63e2ffab14aeab14415069be738d938ebb Signed-off-by: Matthew Bentham <matthew.bentham@arm.com>
Diffstat (limited to 'src/backends/neon/NeonBackend.cpp')
-rw-r--r--src/backends/neon/NeonBackend.cpp8
1 files changed, 7 insertions, 1 deletions
diff --git a/src/backends/neon/NeonBackend.cpp b/src/backends/neon/NeonBackend.cpp
index 66547ad4df..7089f23efa 100644
--- a/src/backends/neon/NeonBackend.cpp
+++ b/src/backends/neon/NeonBackend.cpp
@@ -250,12 +250,18 @@ OptimizationViews NeonBackend::OptimizeSubgraphView(const SubgraphView& subgraph
else if (base.GetType() == LayerType::FullyConnected)
{
FullyConnectedLayer* baseLayer = PolymorphicDowncast<FullyConnectedLayer*>(&base);
+ Optional<TensorInfo> biases;
+
+ if (baseLayer->GetParameters().m_BiasEnabled)
+ {
+ biases = baseLayer->m_Bias->GetTensorInfo();
+ }
arm_compute::Status status = NeonFullyConnectedWorkloadValidate(
baseLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(),
activationLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(),
baseLayer->m_Weight->GetTensorInfo(),
- baseLayer->m_Bias->GetTensorInfo(),
+ biases,
baseLayer->GetParameters(),
&activationDesc);