From 7af00da9068c71ae6f66eaa488048feea62e7aa9 Mon Sep 17 00:00:00 2001 From: jimfly01 Date: Wed, 31 Oct 2018 14:43:53 +0000 Subject: IVGCVSW-2074: Updated CompareDepthwiseConvolution2dTest for NHWC Change-Id: Ib0568024b4f43ae17e6c75ac35d31cf103ce3482 --- src/backends/neon/test/NeonLayerTests.cpp | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) (limited to 'src/backends/neon') diff --git a/src/backends/neon/test/NeonLayerTests.cpp b/src/backends/neon/test/NeonLayerTests.cpp index d242245ba0..8fda825618 100644 --- a/src/backends/neon/test/NeonLayerTests.cpp +++ b/src/backends/neon/test/NeonLayerTests.cpp @@ -439,8 +439,19 @@ ARMNN_AUTO_TEST_CASE(SimpleNormalizationAcrossNhwc, SimpleNormalizationAcrossNhw ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareConv2dWithReference, CompareConvolution2dTest) -ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareDepthwiseConv2dWithReferenceFloat32, CompareDepthwiseConvolution2dTest) -ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareDepthwiseConv2dWithReferenceUint8, CompareDepthwiseConvolution2dTest) +ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareDepthwiseConv2dWithReferenceFloat32, + CompareDepthwiseConvolution2dTest, + armnn::DataLayout::NCHW) +ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareDepthwiseConv2dWithReferenceUint8, + CompareDepthwiseConvolution2dTest, + armnn::DataLayout::NCHW) + +ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareDepthwiseConv2dWithReferenceFloat32Nhwc, + CompareDepthwiseConvolution2dTest, + armnn::DataLayout::NHWC) +ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareDepthwiseConv2dWithReferenceUint8Nhwc, + CompareDepthwiseConvolution2dTest, + armnn::DataLayout::NHWC) ARMNN_COMPARE_REF_AUTO_TEST_CASE(CompareNormalizationWithinWithReference, CompareNormalizationTest, armnn::NormalizationAlgorithmChannel::Within, -- cgit v1.2.1