aboutsummaryrefslogtreecommitdiff
path: root/delegate/src/test/ConvolutionTestHelper.hpp
diff options
context:
space:
mode:
authorJan Eilers <jan.eilers@arm.com>2020-11-20 11:59:40 +0000
committerJim Flynn <jim.flynn@arm.com>2020-11-20 15:58:48 +0000
commiteb61612b0103e44ae3b98e3305b75d01f997662d (patch)
tree0140439dca102411da4d1d872931bd8db688782b /delegate/src/test/ConvolutionTestHelper.hpp
parentf9a2a1e8b5e87bf09bc8010ffba9c3311ce7b649 (diff)
downloadarmnn-eb61612b0103e44ae3b98e3305b75d01f997662d.tar.gz
IVGCVSW-5550 Fix failing delegate Conv2d tests for CpuAcc/GpuAcc
Signed-off-by: Jan Eilers <jan.eilers@arm.com> Change-Id: Ibfd1968a622c2538bbfacf4802cd45096de59db5
Diffstat (limited to 'delegate/src/test/ConvolutionTestHelper.hpp')
-rw-r--r--delegate/src/test/ConvolutionTestHelper.hpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/delegate/src/test/ConvolutionTestHelper.hpp b/delegate/src/test/ConvolutionTestHelper.hpp
index b7705cc904..b3175174d2 100644
--- a/delegate/src/test/ConvolutionTestHelper.hpp
+++ b/delegate/src/test/ConvolutionTestHelper.hpp
@@ -91,7 +91,7 @@ std::vector<char> CreateConv2dTfLiteModel(tflite::BuiltinOperator convolutionOpe
filterQuantizationParameters);
auto biasTensorType = ::tflite::TensorType_FLOAT32;
- if (tensorType == ::tflite::TensorType_UINT8)
+ if (tensorType == ::tflite::TensorType_INT8 || tensorType == ::tflite::TensorType_UINT8)
{
biasTensorType = ::tflite::TensorType_INT32;
}