aboutsummaryrefslogtreecommitdiff
path: root/test/GenericLayerTests.cpp
diff options
context:
space:
mode:
authorDavid Monahan <david.monahan@arm.com>2020-05-19 14:58:34 +0100
committerJan Eilers <jan.eilers@arm.com>2020-05-20 11:43:32 +0100
commitd275cb2fcb3b41928348a408f7eace4ff516ca8b (patch)
tree5a825089eb6c24cebb49b502e6c0404d7b79bd56 /test/GenericLayerTests.cpp
parent813f23049d73177edfc1f1cff71147c39f4b695e (diff)
downloadandroid-nn-driver-d275cb2fcb3b41928348a408f7eace4ff516ca8b.tar.gz
IVGCVSW-IVGCVSW-4772 Fixes for android-nn-driver unit tests on Android R
* Added missing operand parameter numberOfConsumers to DriverTestHelpers * Added nullptr checks to unit tests * Disabling QuantizedLstmTest when ARMCOMPUTECL_ENABLED is undefined Signed-off-by: David Monahan <david.monahan@arm.com> Change-Id: I46dc413d3bb1805eb0e7c982177b6d60d0b297f5
Diffstat (limited to 'test/GenericLayerTests.cpp')
-rw-r--r--test/GenericLayerTests.cpp28
1 files changed, 23 insertions, 5 deletions
diff --git a/test/GenericLayerTests.cpp b/test/GenericLayerTests.cpp
index 961ab166..188c7b1c 100644
--- a/test/GenericLayerTests.cpp
+++ b/test/GenericLayerTests.cpp
@@ -87,9 +87,19 @@ BOOST_AUTO_TEST_CASE(GetSupportedOperations)
// Test Broadcast on add/mul operators
HalPolicy::Model model2 = {};
- AddInputOperand<HalPolicy>(model2, hidl_vec<uint32_t>{1, 1, 3, 4});
- AddInputOperand<HalPolicy>(model2, hidl_vec<uint32_t>{4});
- AddIntOperand<HalPolicy>(model2, actValue);
+ AddInputOperand<HalPolicy>(model2,
+ hidl_vec<uint32_t>{1, 1, 3, 4},
+ HalPolicy::OperandType::TENSOR_FLOAT32,
+ 0.0f,
+ 0,
+ 2);
+ AddInputOperand<HalPolicy>(model2,
+ hidl_vec<uint32_t>{4},
+ HalPolicy::OperandType::TENSOR_FLOAT32,
+ 0.0f,
+ 0,
+ 2);
+ AddIntOperand<HalPolicy>(model2, actValue, 2);
AddOutputOperand<HalPolicy>(model2, hidl_vec<uint32_t>{1, 1, 3, 4});
AddOutputOperand<HalPolicy>(model2, hidl_vec<uint32_t>{1, 1, 3, 4});
@@ -181,8 +191,16 @@ BOOST_AUTO_TEST_CASE(UnsupportedLayerContinueOnFailure)
HalPolicy::OperandType::TENSOR_INT32);
AddInputOperand<HalPolicy>(model,
hidl_vec<uint32_t>{4},
- HalPolicy::OperandType::TENSOR_INT32);
- AddInputOperand<HalPolicy>(model, hidl_vec<uint32_t>{1, 1, 3, 4});
+ HalPolicy::OperandType::TENSOR_INT32,
+ 0.0f,
+ 0,
+ 2);
+ AddInputOperand<HalPolicy>(model,
+ hidl_vec<uint32_t>{1, 1, 3, 4},
+ HalPolicy::OperandType::TENSOR_FLOAT32,
+ 0.0f,
+ 0,
+ 2);
AddOutputOperand<HalPolicy>(model, hidl_vec<uint32_t>{1, 1, 3, 4});
AddOutputOperand<HalPolicy>(model,