diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-11-20 14:50:54 +0000 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2020-11-20 17:35:57 +0000 |
commit | 55518ca7faaf6c2b0cd567afe9fb39d529a10150 (patch) | |
tree | eed61cf0055788a96ab2dc4b16f16d944d2d7ef1 /delegate/src/test/FullyConnectedTestHelper.hpp | |
parent | 63e75dce4d685521f53b26b987d20d11829a4126 (diff) | |
download | armnn-55518ca7faaf6c2b0cd567afe9fb39d529a10150.tar.gz |
IVGCVSW-5544 Fix FullyConnected Delegate tests
* Correct input shape
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I9d1fe4c8ef32a9dfba7f7fdd6af314e9a522fce8
Diffstat (limited to 'delegate/src/test/FullyConnectedTestHelper.hpp')
-rw-r--r-- | delegate/src/test/FullyConnectedTestHelper.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/delegate/src/test/FullyConnectedTestHelper.hpp b/delegate/src/test/FullyConnectedTestHelper.hpp index 4eed9580f1..4b30424d86 100644 --- a/delegate/src/test/FullyConnectedTestHelper.hpp +++ b/delegate/src/test/FullyConnectedTestHelper.hpp @@ -41,7 +41,7 @@ std::vector<char> CreateFullyConnectedTfLiteModel(tflite::TensorType tensorType, 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 }; |