diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-04-26 18:14:12 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2022-05-05 08:49:41 +0000 |
commit | bd22c7d8d71bb9d6fdebcd07a472d66c7616abad (patch) | |
tree | 87a132055db845fab901e18fcb6edd6998e33f3f /delegate/src/test | |
parent | 1299496996bc332f02218f926640a9255ed60310 (diff) | |
download | armnn-bd22c7d8d71bb9d6fdebcd07a472d66c7616abad.tar.gz |
IVGCVSW-6862 Add GATHERNd Neon workload
* Changing the test in the delegate to match one of the unit tests
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I553ca266116ba8ee173fc951ab1ffd2b6eed1428
Diffstat (limited to 'delegate/src/test')
-rw-r--r-- | delegate/src/test/GatherNdTest.cpp | 24 |
1 files changed, 12 insertions, 12 deletions
diff --git a/delegate/src/test/GatherNdTest.cpp b/delegate/src/test/GatherNdTest.cpp index b56a931d27..2b4fd4207e 100644 --- a/delegate/src/test/GatherNdTest.cpp +++ b/delegate/src/test/GatherNdTest.cpp @@ -19,13 +19,13 @@ namespace armnnDelegate void GatherNdUint8Test(std::vector<armnn::BackendId>& backends) { - std::vector<int32_t> paramsShape{8}; - std::vector<int32_t> indicesShape{3,1}; - std::vector<int32_t> expectedOutputShape{3}; + std::vector<int32_t> paramsShape{ 5, 2 }; + std::vector<int32_t> indicesShape{ 3, 1 }; + std::vector<int32_t> expectedOutputShape{ 3, 2 }; - std::vector<uint8_t> paramsValues{1, 2, 3, 4, 5, 6, 7, 8}; - std::vector<int32_t> indicesValues{7, 6, 5}; - std::vector<uint8_t> expectedOutputValues{8, 7, 6}; + std::vector<uint8_t> paramsValues{ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }; + std::vector<int32_t> indicesValues{ 1, 0, 4 }; + std::vector<uint8_t> expectedOutputValues{ 3, 4, 1, 2, 9, 10 }; GatherNdTest<uint8_t>(::tflite::TensorType_UINT8, backends, @@ -39,13 +39,13 @@ void GatherNdUint8Test(std::vector<armnn::BackendId>& backends) void GatherNdFp32Test(std::vector<armnn::BackendId>& backends) { - std::vector<int32_t> paramsShape{8}; - std::vector<int32_t> indicesShape{3,1}; - std::vector<int32_t> expectedOutputShape{3}; + std::vector<int32_t> paramsShape{ 5, 2 }; + std::vector<int32_t> indicesShape{ 3, 1 }; + std::vector<int32_t> expectedOutputShape{ 3, 2 }; - std::vector<float> paramsValues{1.1f, 2.2f, 3.3f, 4.4f, 5.5f, 6.6f, 7.7f, 8.8f}; - std::vector<int32_t> indicesValues{7, 6, 5}; - std::vector<float> expectedOutputValues{8.8f, 7.7f, 6.6f}; + std::vector<float> paramsValues{ 1.1f, 2.2f, 3.3f, 4.4f, 5.5f, 6.6f, 7.7f, 8.8f, 9.9f, 10.10f }; + std::vector<int32_t> indicesValues{ 1, 0, 4 }; + std::vector<float> expectedOutputValues{ 3.3f, 4.4f, 1.1f, 2.2f, 9.9f, 10.10f }; GatherNdTest<float>(::tflite::TensorType_FLOAT32, backends, |