diff options
author | Matteo Martincigh <matteo.martincigh@arm.com> | 2018-10-25 15:39:33 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2018-10-25 16:57:26 +0100 |
commit | 8d50f8f4b848547635a74e6fdec29f632748ebb1 (patch) | |
tree | 50b2c00fa2a9e7a4e1dc24326e39a3338f1ec296 /test/DriverTestHelpers.hpp | |
parent | 50db26cbd83e7753c1581cc8c29b8e575da66ade (diff) | |
download | android-nn-driver-8d50f8f4b848547635a74e6fdec29f632748ebb1.tar.gz |
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
Diffstat (limited to 'test/DriverTestHelpers.hpp')
-rw-r--r-- | test/DriverTestHelpers.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
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; |