diff options
author | Nikhil Raj <nikhil.raj@arm.com> | 2018-09-03 11:25:56 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-09-18 12:40:40 +0100 |
commit | 77605826a353981d41f0ee346850d411770535f8 (patch) | |
tree | 5cdd4c8cb6768d7651089105ee2884f536d8a34c /test/FullyConnected.cpp | |
parent | b0717b5241a15e3e4d37a1b51b6e5fd9a92a664f (diff) | |
download | android-nn-driver-77605826a353981d41f0ee346850d411770535f8.tar.gz |
IVGCVSW-1713 Create a minimum unit test to compare the results
before and after passing the FP16 flag in the Android-nn-driver
Change-Id: If8d4ca12421c3bee2526eec98f11d393af822373
Diffstat (limited to 'test/FullyConnected.cpp')
-rw-r--r-- | test/FullyConnected.cpp | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/test/FullyConnected.cpp b/test/FullyConnected.cpp index 85e58d02..5c204ca7 100644 --- a/test/FullyConnected.cpp +++ b/test/FullyConnected.cpp @@ -19,7 +19,7 @@ BOOST_AUTO_TEST_CASE(FullyConnected) // but that uses slightly weird dimensions which I don't think we need to support for now auto driver = std::make_unique<ArmnnDriver>(DriverOptions(armnn::Compute::CpuRef)); - neuralnetworks::V1_0::Model model = {}; + V1_0::Model model = {}; // add operands int32_t actValue = 0; @@ -34,7 +34,7 @@ BOOST_AUTO_TEST_CASE(FullyConnected) // make the fully connected operation model.operations.resize(1); - model.operations[0].type = neuralnetworks::V1_0::OperationType::FULLY_CONNECTED; + model.operations[0].type = V1_0::OperationType::FULLY_CONNECTED; model.operations[0].inputs = hidl_vec<uint32_t>{0, 1, 2, 3}; model.operations[0].outputs = hidl_vec<uint32_t>{4}; @@ -90,7 +90,7 @@ BOOST_AUTO_TEST_CASE(TestFullyConnected4dInput) sup = supported; }; - neuralnetworks::V1_0::Model model = {}; + V1_0::Model model = {}; // operands int32_t actValue = 0; @@ -113,7 +113,7 @@ BOOST_AUTO_TEST_CASE(TestFullyConnected4dInput) model.operations.resize(1); - model.operations[0].type = neuralnetworks::V1_0::OperationType::FULLY_CONNECTED; + model.operations[0].type = V1_0::OperationType::FULLY_CONNECTED; model.operations[0].inputs = hidl_vec<uint32_t>{0,1,2,3}; model.operations[0].outputs = hidl_vec<uint32_t>{4}; @@ -177,7 +177,7 @@ BOOST_AUTO_TEST_CASE(TestFullyConnected4dInputReshape) sup = supported; }; - neuralnetworks::V1_0::Model model = {}; + V1_0::Model model = {}; // operands int32_t actValue = 0; @@ -200,7 +200,7 @@ BOOST_AUTO_TEST_CASE(TestFullyConnected4dInputReshape) model.operations.resize(1); - model.operations[0].type = neuralnetworks::V1_0::OperationType::FULLY_CONNECTED; + model.operations[0].type = V1_0::OperationType::FULLY_CONNECTED; model.operations[0].inputs = hidl_vec<uint32_t>{0,1,2,3}; model.operations[0].outputs = hidl_vec<uint32_t>{4}; |