diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-04-01 16:51:23 +0100 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2020-04-01 21:57:05 +0000 |
commit | e8cddebd26a1dc16b713e4aea1041fb9c8944ea7 (patch) | |
tree | bf153ff5b8d733b10152c08e0c42c8c9ac8df867 /src/backends/backendsCommon/test/layerTests/Conv2dTestImpl.hpp | |
parent | db73c98f491d2fb17991acc13621c80cad628fb0 (diff) | |
download | armnn-e8cddebd26a1dc16b713e4aea1041fb9c8944ea7.tar.gz |
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 <narumol.prangnawarat@arm.com>
Change-Id: Ia59c92a22001b41f2681dafb951203a2223d3367
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/Conv2dTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/layerTests/Conv2dTestImpl.hpp | 12 |
1 files changed, 12 insertions, 0 deletions
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<uint8_t, 4> Convolution2dPerAxisQuantTest( const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, const armnn::DataLayout layout); +LayerTestResult<float, 4> Convolution2d3x3Stride2x2BFloat16Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + bool biasEnabled, + const armnn::DataLayout& dataLayout); + +LayerTestResult<float, 4> Convolution2d3x3Stride2x2BFloat16SmallValueTest( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + bool biasEnabled, + const armnn::DataLayout& dataLayout); + // // DepthwiseConvolution2d // |