From e8cddebd26a1dc16b713e4aea1041fb9c8944ea7 Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Wed, 1 Apr 2020 16:51:23 +0100 Subject: IVGCVSW-4643 Add Convolution2D tests with Bfloat16 input and Float32 output * Modify SimpleConvolution2dNhwcTestImpl to allow different input and out types * Add unit tests for Conv2D with Bfloat16 input and Float32 output to compare the accuracy with Float32 for small value inputs and big value inputs Signed-off-by: Narumol Prangnawarat Change-Id: Ia59c92a22001b41f2681dafb951203a2223d3367 --- .../backendsCommon/test/layerTests/Conv2dTestImpl.hpp | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'src/backends/backendsCommon/test/layerTests/Conv2dTestImpl.hpp') diff --git a/src/backends/backendsCommon/test/layerTests/Conv2dTestImpl.hpp b/src/backends/backendsCommon/test/layerTests/Conv2dTestImpl.hpp index 6261bdfca6..21c4cf9bf0 100644 --- a/src/backends/backendsCommon/test/layerTests/Conv2dTestImpl.hpp +++ b/src/backends/backendsCommon/test/layerTests/Conv2dTestImpl.hpp @@ -116,6 +116,18 @@ LayerTestResult Convolution2dPerAxisQuantTest( const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, const armnn::DataLayout layout); +LayerTestResult Convolution2d3x3Stride2x2BFloat16Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + bool biasEnabled, + const armnn::DataLayout& dataLayout); + +LayerTestResult Convolution2d3x3Stride2x2BFloat16SmallValueTest( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + bool biasEnabled, + const armnn::DataLayout& dataLayout); + // // DepthwiseConvolution2d // -- cgit v1.2.1