MLCE-160 Error loading quantized model containing BatchNorm Layer
* Relaxed restrictions in BatchNormalizationQueueDescriptor::Validate
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I3101971c2101e90144bbbf7b63367cb0ef09573f
diff --git a/src/backends/backendsCommon/test/WorkloadDataValidation.cpp b/src/backends/backendsCommon/test/WorkloadDataValidation.cpp
index 5c60e9e..d48a2bb 100644
--- a/src/backends/backendsCommon/test/WorkloadDataValidation.cpp
+++ b/src/backends/backendsCommon/test/WorkloadDataValidation.cpp
@@ -19,7 +19,31 @@
BOOST_AUTO_TEST_SUITE(WorkloadInfoValidation)
+BOOST_AUTO_TEST_CASE(BatchNormalizationQueueDescriptor_Validate_DifferentQuantizationData)
+{
+ TensorShape inputShape { 1, 3, 2, 2 };
+ TensorShape outputShape { 1, 3, 2, 2 };
+ TensorInfo inputTensorInfo(inputShape, armnn::DataType::QAsymmU8, .1f, 125);
+ TensorInfo outputTensorInfo(outputShape, armnn::DataType::QAsymmU8, .2f, 120);
+
+ BatchNormalizationQueueDescriptor invalidData;
+ WorkloadInfo invalidInfo;
+
+ unsigned int sameShape[] = { 10 };
+ TensorInfo sameInfo = armnn::TensorInfo(1, sameShape, armnn::DataType::QAsymmU8);
+ ScopedCpuTensorHandle sameTensor(sameInfo);
+
+ AddInputToWorkload(invalidData, invalidInfo, inputTensorInfo, nullptr);
+ AddOutputToWorkload(invalidData, invalidInfo, outputTensorInfo, nullptr);
+
+ invalidData.m_Mean = &sameTensor;
+ invalidData.m_Variance = &sameTensor;
+ invalidData.m_Beta= &sameTensor;
+ invalidData.m_Gamma = &sameTensor;
+
+ BOOST_CHECK_NO_THROW(RefBatchNormalizationWorkload(invalidData, invalidInfo));
+}
BOOST_AUTO_TEST_CASE(QueueDescriptor_Validate_WrongNumOfInputsOutputs)
{