diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2019-06-24 14:24:36 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2019-06-25 08:48:10 +0000 |
commit | 57f13d5905a1fbdc89b53d68b2bdc6b753b9e8d5 (patch) | |
tree | a982eecfcb9cde7c8326034f7af4a333a2ab66b8 /src/backends/reference/test/RefCreateWorkloadTests.cpp | |
parent | 34757810f8b734f5f59485a542b56934ad4cc5f0 (diff) | |
download | armnn-57f13d5905a1fbdc89b53d68b2bdc6b753b9e8d5.tar.gz |
IVGCVSW-3334 Refactor BatchToSpace tests to be generic
* Generify and reformat test BatchToSpace tests
* Add missing RefCreateWorkload test
Change-Id: I08af018c07ee41df5b9d1e578d99bc03f38090ac
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Diffstat (limited to 'src/backends/reference/test/RefCreateWorkloadTests.cpp')
-rw-r--r-- | src/backends/reference/test/RefCreateWorkloadTests.cpp | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefCreateWorkloadTests.cpp b/src/backends/reference/test/RefCreateWorkloadTests.cpp index d174093fd4..67b1d945db 100644 --- a/src/backends/reference/test/RefCreateWorkloadTests.cpp +++ b/src/backends/reference/test/RefCreateWorkloadTests.cpp @@ -691,6 +691,29 @@ BOOST_AUTO_TEST_CASE(CreateRsqrtQsymm16) RefCreateRsqrtTest<RefRsqrtWorkload, armnn::DataType::QuantisedSymm16>(); } +template <typename BatchToSpaceNdWorkloadType, armnn::DataType DataType> +static void RefCreateBatchToSpaceNdTest() +{ + Graph graph; + RefWorkloadFactory factory; + + auto workload = CreateBatchToSpaceNdWorkloadTest<BatchToSpaceNdWorkloadType, DataType>(factory, graph); + + CheckInputOutput(std::move(workload), + TensorInfo({ 1, 1 }, DataType), + TensorInfo({ 1, 1 }, DataType)); +} + +BOOST_AUTO_TEST_CASE(CreateBatchToSpaceNdFloat32) +{ + RefCreateBatchToSpaceNdTest<RefBatchToSpaceNdWorkload, armnn::DataType::Float32>(); +} + +BOOST_AUTO_TEST_CASE(CreateBatchToSpaceNdUint8) +{ + RefCreateBatchToSpaceNdTest<RefBatchToSpaceNdWorkload, armnn::DataType::QuantisedAsymm8>(); +} + template <typename L2NormalizationWorkloadType, armnn::DataType DataType> static void RefCreateL2NormalizationTest(DataLayout dataLayout) { |