From 67d6390c3b37c83bad2180ee4dc54893143c7b29 Mon Sep 17 00:00:00 2001 From: Matthew Bentham Date: Tue, 8 Feb 2022 15:03:07 +0000 Subject: 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 --- src/backends/neon/workloads/NeonFullyConnectedWorkload.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/backends/neon/workloads/NeonFullyConnectedWorkload.hpp') diff --git a/src/backends/neon/workloads/NeonFullyConnectedWorkload.hpp b/src/backends/neon/workloads/NeonFullyConnectedWorkload.hpp index b5f616057f..419a3299f2 100644 --- a/src/backends/neon/workloads/NeonFullyConnectedWorkload.hpp +++ b/src/backends/neon/workloads/NeonFullyConnectedWorkload.hpp @@ -20,7 +20,7 @@ namespace armnn arm_compute::Status NeonFullyConnectedWorkloadValidate(const TensorInfo& input, const TensorInfo& output, const TensorInfo& weights, - const TensorInfo& biases, + const Optional& biases, const FullyConnectedDescriptor& descriptor, const ActivationDescriptor* activationDescriptor = nullptr); -- cgit v1.2.1