From 8d50f8f4b848547635a74e6fdec29f632748ebb1 Mon Sep 17 00:00:00 2001 From: Matteo Martincigh Date: Thu, 25 Oct 2018 15:39:33 +0100 Subject: IVGCVSW-2051 Added unit tests in the Android NN Driver for the new Mean layer * Added unit tests (in a new file test/1.1/Mean.cpp) * Refactored ArmnnDriverImpl to remove redundant tags in the log messages * Modified AddInputOperand to accept a quantized input tensor Change-Id: Ie037ce426777daab28b0501124e1cc8686ad6184 --- test/DriverTestHelpers.hpp | 1 + 1 file changed, 1 insertion(+) (limited to 'test/DriverTestHelpers.hpp') diff --git a/test/DriverTestHelpers.hpp b/test/DriverTestHelpers.hpp index cce220e5..370936fe 100644 --- a/test/DriverTestHelpers.hpp +++ b/test/DriverTestHelpers.hpp @@ -146,6 +146,7 @@ void AddInputOperand(HalModel& model, { Operand op = {}; op.type = operandType; + op.scale = operandType == OperandType::TENSOR_QUANT8_ASYMM ? 1.f / 255.f : 0.f; op.dimensions = dimensions; op.lifetime = OperandLifeTime::MODEL_INPUT; -- cgit v1.2.1