diff options
| author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-07-02 17:25:47 +0100 |
|---|---|---|
| committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-07-12 14:44:27 +0000 |
| commit | cc0cefb6ca2abf22fc0128548b93a492df530705 (patch) | |
| tree | 3b87a05e2b83ee432a04b7c43f769e0a572a2a5c /src/backends/cl/test/ClCreateWorkloadTests.cpp | |
| parent | 59c6670cac0aac1fedf426fb19ccd578da6f9f55 (diff) | |
| download | armnn-cc0cefb6ca2abf22fc0128548b93a492df530705.tar.gz | |
IVGCVSW-3296 Add CL backend support for ResizeNearestNeighbour
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I7f4c722141837939fd8904c52e75704a15c8a5e3
Diffstat (limited to 'src/backends/cl/test/ClCreateWorkloadTests.cpp')
| -rw-r--r-- | src/backends/cl/test/ClCreateWorkloadTests.cpp | 33 |
1 files changed, 21 insertions, 12 deletions
diff --git a/src/backends/cl/test/ClCreateWorkloadTests.cpp b/src/backends/cl/test/ClCreateWorkloadTests.cpp index b89abdb473..de13390f87 100644 --- a/src/backends/cl/test/ClCreateWorkloadTests.cpp +++ b/src/backends/cl/test/ClCreateWorkloadTests.cpp @@ -758,16 +758,15 @@ BOOST_AUTO_TEST_CASE(CreateLSTMWorkloadFloatWorkload) ClCreateLstmWorkloadTest<ClLstmFloatWorkload>(); } -template <typename ResizeBilinearWorkloadType, typename armnn::DataType DataType> -static void ClResizeBilinearWorkloadTest(DataLayout dataLayout) +template <typename ResizeWorkloadType, typename armnn::DataType DataType> +static void ClResizeWorkloadTest(DataLayout dataLayout) { Graph graph; ClWorkloadFactory factory = ClWorkloadFactoryHelper::GetFactory(ClWorkloadFactoryHelper::GetMemoryManager()); - auto workload = CreateResizeBilinearWorkloadTest<ResizeBilinearWorkloadType, DataType>(factory, graph, dataLayout); + auto workload = CreateResizeBilinearWorkloadTest<ResizeWorkloadType, DataType>(factory, graph, dataLayout); - // Checks that inputs/outputs are as we expect them (see definition of CreateResizeBilinearWorkloadTest). auto queueDescriptor = workload->GetData(); auto inputHandle = boost::polymorphic_downcast<IClTensorHandle*>(queueDescriptor.m_Inputs[0]); @@ -786,24 +785,34 @@ static void ClResizeBilinearWorkloadTest(DataLayout dataLayout) } } -BOOST_AUTO_TEST_CASE(CreateResizeBilinearFloat32NchwWorkload) +BOOST_AUTO_TEST_CASE(CreateResizeFloat32NchwWorkload) { - ClResizeBilinearWorkloadTest<ClResizeBilinearFloatWorkload, armnn::DataType::Float32>(DataLayout::NCHW); + ClResizeWorkloadTest<ClResizeWorkload, armnn::DataType::Float32>(DataLayout::NCHW); } -BOOST_AUTO_TEST_CASE(CreateResizeBilinearFloat16NchwWorkload) +BOOST_AUTO_TEST_CASE(CreateResizeFloat16NchwWorkload) { - ClResizeBilinearWorkloadTest<ClResizeBilinearFloatWorkload, armnn::DataType::Float16>(DataLayout::NCHW); + ClResizeWorkloadTest<ClResizeWorkload, armnn::DataType::Float16>(DataLayout::NCHW); } -BOOST_AUTO_TEST_CASE(CreateResizeBilinearFloat32NhwcWorkload) +BOOST_AUTO_TEST_CASE(CreateResizeUint8NchwWorkload) { - ClResizeBilinearWorkloadTest<ClResizeBilinearFloatWorkload, armnn::DataType::Float32>(DataLayout::NHWC); + ClResizeWorkloadTest<ClResizeWorkload, armnn::DataType::QuantisedAsymm8>(DataLayout::NCHW); } -BOOST_AUTO_TEST_CASE(CreateResizeBilinearFloat16NhwcWorkload) +BOOST_AUTO_TEST_CASE(CreateResizeFloat32NhwcWorkload) { - ClResizeBilinearWorkloadTest<ClResizeBilinearFloatWorkload, armnn::DataType::Float16>(DataLayout::NHWC); + ClResizeWorkloadTest<ClResizeWorkload, armnn::DataType::Float32>(DataLayout::NHWC); +} + +BOOST_AUTO_TEST_CASE(CreateResizeFloat16NhwcWorkload) +{ + ClResizeWorkloadTest<ClResizeWorkload, armnn::DataType::Float16>(DataLayout::NHWC); +} + +BOOST_AUTO_TEST_CASE(CreateResizeUint8NhwcWorkload) +{ + ClResizeWorkloadTest<ClResizeWorkload, armnn::DataType::QuantisedAsymm8>(DataLayout::NHWC); } template <typename MeanWorkloadType, typename armnn::DataType DataType> |