diff options
author | James Conroy <james.conroy@arm.com> | 2018-11-01 11:33:09 +0000 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2018-11-01 15:45:50 +0000 |
commit | 6b9658239d377372523fe49c71fde31701d986e3 (patch) | |
tree | 0b14b3e81a1321c8e5e85d2800a6e969d6cb724f /src/backends/test/LayerTests.hpp | |
parent | b9c8963c3d393baf27edf37ab732fa76ee53af50 (diff) | |
download | armnn-6b9658239d377372523fe49c71fde31701d986e3.tar.gz |
IVGCVSW-2103: Add 2-Channel unit tests ResizeBilinear
* Modifies ResizeBilinear unit tests to use 2-Channel
tensor shapes for input and output data, to improve
test coverage when exercising NHWC data layout.
* Refactors unit tests to permute input and output
data when exercising NHWC data layout.
Change-Id: Ib7fb438cac23e78ff0104c895c3b7596bf7c3aa7
Diffstat (limited to 'src/backends/test/LayerTests.hpp')
-rw-r--r-- | src/backends/test/LayerTests.hpp | 22 |
1 files changed, 10 insertions, 12 deletions
diff --git a/src/backends/test/LayerTests.hpp b/src/backends/test/LayerTests.hpp index 39cd4c4a0b..26dec60e0b 100644 --- a/src/backends/test/LayerTests.hpp +++ b/src/backends/test/LayerTests.hpp @@ -253,27 +253,25 @@ LayerTestResult<float, 4> CompareBoundedReLuTest(armnn::IWorkloadFactory& worklo float lowerBound); // Tests that the output should be identical to the input when the output dimensions match the input ones. -LayerTestResult<float, 4> ResizeBilinearNopTest(armnn::IWorkloadFactory& workloadFactory); +LayerTestResult<float, 4> ResizeBilinearNopTest(armnn::IWorkloadFactory& workloadFactory, + const armnn::DataLayoutIndexed& dataLayout); // Tests the behaviour of the resize bilinear operation when rescaling a 2x2 image into a 1x1 image. -LayerTestResult<float, 4> SimpleResizeBilinearTest(armnn::IWorkloadFactory& workloadFactory); +LayerTestResult<float, 4> SimpleResizeBilinearTest(armnn::IWorkloadFactory& workloadFactory, + const armnn::DataLayoutIndexed& dataLayout); // Tests the resize bilinear for minification of a square input matrix (also: input dimensions are a // multiple of output dimensions). -LayerTestResult<float, 4> ResizeBilinearSqMinTest(armnn::IWorkloadFactory& workloadFactory); +LayerTestResult<float, 4> ResizeBilinearSqMinTest(armnn::IWorkloadFactory& workloadFactory, + const armnn::DataLayoutIndexed& dataLayout); // Tests the resize bilinear for minification (output dimensions smaller than input dimensions). -LayerTestResult<float, 4> ResizeBilinearMinTest(armnn::IWorkloadFactory& workloadFactory); +LayerTestResult<float, 4> ResizeBilinearMinTest(armnn::IWorkloadFactory& workloadFactory, + const armnn::DataLayoutIndexed& dataLayout); // Tests the resize bilinear for magnification (output dimensions bigger than input dimensions). -LayerTestResult<float, 4> ResizeBilinearMagTest(armnn::IWorkloadFactory& workloadFactory); - -// Tests that execute Resize Bilinear with NHWC data layout -LayerTestResult<float, 4> ResizeBilinearNopNhwcTest(armnn::IWorkloadFactory& workloadFactory); -LayerTestResult<float, 4> SimpleResizeBilinearNhwcTest(armnn::IWorkloadFactory& workloadFactory); -LayerTestResult<float, 4> ResizeBilinearSqMinNhwcTest(armnn::IWorkloadFactory& workloadFactory); -LayerTestResult<float, 4> ResizeBilinearMinNhwcTest(armnn::IWorkloadFactory& workloadFactory); -LayerTestResult<float, 4> ResizeBilinearMagNhwcTest(armnn::IWorkloadFactory& workloadFactory); +LayerTestResult<float, 4> ResizeBilinearMagTest(armnn::IWorkloadFactory& workloadFactory, + const armnn::DataLayoutIndexed& dataLayout); LayerTestResult<float, 4> BatchNormTest(armnn::IWorkloadFactory& workloadFactory); LayerTestResult<float, 4> BatchNormNhwcTest(armnn::IWorkloadFactory& workloadFactory); |