diff options
author | jimfly01 <jim.flynn@arm.com> | 2018-10-31 14:43:53 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2018-11-01 12:13:37 +0000 |
commit | 7af00da9068c71ae6f66eaa488048feea62e7aa9 (patch) | |
tree | 1c0808b0a1beeb85a3103f8fe01d561b1011ecc5 /src/backends/test/LayerTests.cpp | |
parent | 1b417c4970462d08e8e80d2f2462230ca5b91515 (diff) | |
download | armnn-7af00da9068c71ae6f66eaa488048feea62e7aa9.tar.gz |
IVGCVSW-2074: Updated CompareDepthwiseConvolution2dTest for NHWC
Change-Id: Ib0568024b4f43ae17e6c75ac35d31cf103ce3482
Diffstat (limited to 'src/backends/test/LayerTests.cpp')
-rwxr-xr-x | src/backends/test/LayerTests.cpp | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/backends/test/LayerTests.cpp b/src/backends/test/LayerTests.cpp index abe3704a17..49c6d30b6d 100755 --- a/src/backends/test/LayerTests.cpp +++ b/src/backends/test/LayerTests.cpp @@ -683,15 +683,16 @@ LayerTestResult<float,4> CompareConvolution2dTest(armnn::IWorkloadFactory& workl template<typename T> LayerTestResult<T,4> CompareDepthwiseConvolution2dTest(armnn::IWorkloadFactory& workloadFactory, - armnn::IWorkloadFactory& refWorkloadFactory) + armnn::IWorkloadFactory& refWorkloadFactory, + const armnn::DataLayoutIndexed& layout) { - return CompareDepthwiseConvolution2dTestImpl<T>(workloadFactory, refWorkloadFactory); + return CompareDepthwiseConvolution2dTestImpl<T>(workloadFactory, refWorkloadFactory, layout); } template LayerTestResult<float, 4> CompareDepthwiseConvolution2dTest<float>( - armnn::IWorkloadFactory&, armnn::IWorkloadFactory&); + armnn::IWorkloadFactory&, armnn::IWorkloadFactory&, const armnn::DataLayoutIndexed&); template LayerTestResult<uint8_t, 4> CompareDepthwiseConvolution2dTest<uint8_t>( - armnn::IWorkloadFactory&, armnn::IWorkloadFactory&); + armnn::IWorkloadFactory&, armnn::IWorkloadFactory&, const armnn::DataLayoutIndexed&); LayerTestResult<float,4> SimpleNormalizationAcrossTest(armnn::IWorkloadFactory& workloadFactory) { |