diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-07-28 10:42:13 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-07-28 16:45:05 +0000 |
commit | d109a4d0fb980f2e97c05cd86451eda0b53f37ef (patch) | |
tree | 19092204016ccd60ea36637cc06a48a224508570 /src/armnnTfLiteParser/test/FullyConnected.cpp | |
parent | 2f746b3f346d3efa9071fc53592652425869d6b3 (diff) | |
download | armnn-d109a4d0fb980f2e97c05cd86451eda0b53f37ef.tar.gz |
IVGCVSW-4932 Introduce ShapeInferenceMethod to TfLite Parser
* Introduced ShapeInferenceMethod::InferAndValidate to TfLiteParser
* Added unit tests
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Iad4aadce92912e7645b1652019ec4af478b7fc32
Diffstat (limited to 'src/armnnTfLiteParser/test/FullyConnected.cpp')
-rw-r--r-- | src/armnnTfLiteParser/test/FullyConnected.cpp | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/src/armnnTfLiteParser/test/FullyConnected.cpp b/src/armnnTfLiteParser/test/FullyConnected.cpp index d1223d5af2..e7aa9082e2 100644 --- a/src/armnnTfLiteParser/test/FullyConnected.cpp +++ b/src/armnnTfLiteParser/test/FullyConnected.cpp @@ -171,4 +171,28 @@ BOOST_FIXTURE_TEST_CASE(FullyConnectedWithBiasMultipleOutputs, FullyConnectedWit { (40+10)/2, (400+10)/2 }); } +struct DynamicFullyConnectedWithBiasMultipleOutputsFixture : FullyConnectedFixture +{ + DynamicFullyConnectedWithBiasMultipleOutputsFixture() + : FullyConnectedFixture("[ 1, 4, 2, 1 ]", // inputShape + "[ ]", // outputShape + "[ 1, 4 ]", // filterShape + "[ 2, 3, 4, 5 ]", // filterData + "[ 1 ]", // biasShape + "[ 10, 0, 0, 0 ]" ) // biasData + { } +}; + +BOOST_FIXTURE_TEST_CASE( + DynamicFullyConnectedWithBiasMultipleOutputs, + DynamicFullyConnectedWithBiasMultipleOutputsFixture) +{ + RunTest<2, + armnn::DataType::QAsymmU8, + armnn::DataType::QAsymmU8>(0, + { { "inputTensor", { 1, 2, 3, 4, 10, 20, 30, 40} } }, + { { "outputTensor", { (40+10)/2, (400+10)/2 } } }, + true); +} + BOOST_AUTO_TEST_SUITE_END() |