diff options
author | surmeh01 <surabhi.mehta@arm.com> | 2018-05-18 16:31:43 +0100 |
---|---|---|
committer | telsoa01 <telmo.soares@arm.com> | 2018-05-23 13:09:07 +0100 |
commit | 3537c2ca7ebf31c1673b9ec2bb0c17b0406bbae0 (patch) | |
tree | 5950603ad78ec3fe56fb31ddc7f4d52a19f5bc60 /src/armnn/backends/RefWorkloads/ConvImpl.hpp | |
parent | bceff2fb3fc68bb0aa88b886900c34b77340c826 (diff) | |
download | armnn-3537c2ca7ebf31c1673b9ec2bb0c17b0406bbae0.tar.gz |
Release 18.05
Diffstat (limited to 'src/armnn/backends/RefWorkloads/ConvImpl.hpp')
-rw-r--r-- | src/armnn/backends/RefWorkloads/ConvImpl.hpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/armnn/backends/RefWorkloads/ConvImpl.hpp b/src/armnn/backends/RefWorkloads/ConvImpl.hpp index ecc5b14687..8b66b0b7d2 100644 --- a/src/armnn/backends/RefWorkloads/ConvImpl.hpp +++ b/src/armnn/backends/RefWorkloads/ConvImpl.hpp @@ -57,6 +57,11 @@ static void ConvImpl(ConvData data, int32_t outputOffset, bool depthwise = false) { + if (data.m_Parameters.m_BiasEnabled && !biasData) + { + throw InvalidArgumentException("Bias is enabled but the bias data is invalid"); + } + const TensorInfo& inputInfo0 = GetTensorInfo(data.m_Inputs[0]); const TensorInfo& outputInfo0 = GetTensorInfo(data.m_Outputs[0]); const TensorInfo& filterInfo = data.m_Weight->GetTensorInfo(); @@ -65,8 +70,6 @@ static void ConvImpl(ConvData data, unsigned int channelsInput = filterInfo.GetShape()[1]; unsigned int channelsOutput = depthwise ? channelsInput * depthMult : filterInfo.GetShape()[0]; - BOOST_ASSERT(data.m_Parameters.m_BiasEnabled == false || biasData != nullptr); - unsigned int batchSize = outputInfo0.GetShape()[0]; unsigned int heightOutput = outputInfo0.GetShape()[2]; unsigned int widthOutput = outputInfo0.GetShape()[3]; |