diff options
Diffstat (limited to 'src/backends/cl/workloads')
| -rw-r--r-- | src/backends/cl/workloads/ClFullyConnectedWorkload.cpp | 5 | ||||
| -rw-r--r-- | src/backends/cl/workloads/ClFullyConnectedWorkload.hpp | 2 |
2 files changed, 4 insertions, 3 deletions
diff --git a/src/backends/cl/workloads/ClFullyConnectedWorkload.cpp b/src/backends/cl/workloads/ClFullyConnectedWorkload.cpp index 3eb53e64b4..017f4fff6b 100644 --- a/src/backends/cl/workloads/ClFullyConnectedWorkload.cpp +++ b/src/backends/cl/workloads/ClFullyConnectedWorkload.cpp @@ -19,7 +19,7 @@ using namespace armcomputetensorutils; arm_compute::Status ClFullyConnectedWorkloadValidate(const TensorInfo& input, const TensorInfo& output, const TensorInfo& weights, - const TensorInfo& biases, + const Optional<TensorInfo>& biases, const FullyConnectedDescriptor& descriptor, const ActivationDescriptor* activationDescriptor) { @@ -31,7 +31,8 @@ arm_compute::Status ClFullyConnectedWorkloadValidate(const TensorInfo& input, arm_compute::TensorInfo* optionalAclBiases = nullptr; if (descriptor.m_BiasEnabled) { - aclBiases = BuildArmComputeTensorInfo(biases); + ARMNN_ASSERT(biases.has_value()); + aclBiases = BuildArmComputeTensorInfo(biases.value()); optionalAclBiases = &aclBiases; } diff --git a/src/backends/cl/workloads/ClFullyConnectedWorkload.hpp b/src/backends/cl/workloads/ClFullyConnectedWorkload.hpp index 210757779f..3ab9f986a8 100644 --- a/src/backends/cl/workloads/ClFullyConnectedWorkload.hpp +++ b/src/backends/cl/workloads/ClFullyConnectedWorkload.hpp @@ -18,7 +18,7 @@ namespace armnn arm_compute::Status ClFullyConnectedWorkloadValidate(const TensorInfo& input, const TensorInfo& output, const TensorInfo& weights, - const TensorInfo& biases, + const Optional<TensorInfo>& biases, const FullyConnectedDescriptor& descriptor, const ActivationDescriptor* activationDescriptor = nullptr); |