commit 55518ca7faaf6c2b0cd567afe9fb39d529a10150
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Fri Nov 20 14:50:54 2020 +0000
committer Francis Murtagh <francis.murtagh@arm.com> Fri Nov 20 17:35:57 2020 +0000
tree eed61cf0055788a96ab2dc4b16f16d944d2d7ef1
parent 63e75dce4d685521f53b26b987d20d11829a4126

IVGCVSW-5544 Fix FullyConnected Delegate tests

 * Correct input shape

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I9d1fe4c8ef32a9dfba7f7fdd6af314e9a522fce8

delegate/src/test/FullyConnectedTestHelper.hpp

diff --git a/delegate/src/test/FullyConnectedTestHelper.hpp b/delegate/src/test/FullyConnectedTestHelper.hpp
index 4eed958..4b30424 100644
--- a/delegate/src/test/FullyConnectedTestHelper.hpp
+++ b/delegate/src/test/FullyConnectedTestHelper.hpp
@@ -41,7 +41,7 @@
                                                     sizeof(T) * weightsData.size()));
 
     auto biasTensorType = ::tflite::TensorType_FLOAT32;
-    if (tensorType == ::tflite::TensorType_UINT8)
+    if (tensorType == ::tflite::TensorType_INT8)
     {
         biasTensorType = ::tflite::TensorType_INT32;
         std::vector<int32_t> biasData = { 10 };