diff options
author | nikraj01 <nikhil.raj@arm.com> | 2019-06-04 15:35:34 +0100 |
---|---|---|
committer | nikraj01 <nikhil.raj@arm.com> | 2019-06-04 15:35:34 +0100 |
commit | 33732f64df7d3ed014354806f1b855014501be05 (patch) | |
tree | b3ce07e6cfa731bc744563b5c0e97722911603ba /src/backends/reference/test/RefCreateWorkloadTests.cpp | |
parent | b40d710508aede83fd1cbc17b4890c111fe0b297 (diff) | |
download | armnn-33732f64df7d3ed014354806f1b855014501be05.tar.gz |
IVGCVSW-3215 Add CreateWorkload test for Rsqrt Fp32
Change-Id: Iea6edf90148773b12f361bfa4b1d572c5322d18b
Signed-off-by: nikraj01 <nikhil.raj@arm.com>
Diffstat (limited to 'src/backends/reference/test/RefCreateWorkloadTests.cpp')
-rw-r--r-- | src/backends/reference/test/RefCreateWorkloadTests.cpp | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefCreateWorkloadTests.cpp b/src/backends/reference/test/RefCreateWorkloadTests.cpp index 0311276f10..7c5712b915 100644 --- a/src/backends/reference/test/RefCreateWorkloadTests.cpp +++ b/src/backends/reference/test/RefCreateWorkloadTests.cpp @@ -614,6 +614,26 @@ BOOST_AUTO_TEST_CASE(CreateResizeBilinearFloat32Nhwc) RefCreateResizeBilinearTest<RefResizeBilinearFloat32Workload, armnn::DataType::Float32>(DataLayout::NHWC); } +template <typename RsqrtWorkloadType, armnn::DataType DataType> +static void RefCreateRsqrtTest() +{ + Graph graph; + RefWorkloadFactory factory; + + auto workload = CreateRsqrtWorkloadTest<RsqrtWorkloadType, DataType>(factory, graph); + + // Checks that outputs are as we expect them (see definition of CreateRsqrtWorkloadTest). + CheckInputOutput(std::move(workload), + TensorInfo({ 1, 1 }, DataType), + TensorInfo({ 1, 1 }, DataType)); + +} + +BOOST_AUTO_TEST_CASE(CreateRsqrtFloat32) +{ + RefCreateRsqrtTest<RefRsqrtFloat32Workload, armnn::DataType::Float32>(); +} + template <typename L2NormalizationWorkloadType, armnn::DataType DataType> static void RefCreateL2NormalizationTest(DataLayout dataLayout) { |