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/reference | |
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/reference')
-rw-r--r-- | src/backends/reference/test/RefLayerTests.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefLayerTests.cpp b/src/backends/reference/test/RefLayerTests.cpp index 1c96db1c37..6bcd7e3ed5 100644 --- a/src/backends/reference/test/RefLayerTests.cpp +++ b/src/backends/reference/test/RefLayerTests.cpp @@ -172,6 +172,12 @@ ARMNN_AUTO_TEST_CASE(Convolution2d2x2Dilation2x2Padding2x2Stride3x3NhwcInt16, ARMNN_AUTO_TEST_CASE(Convolution2dPerAxisQuantTestNchw, Convolution2dPerAxisQuantTest, DataLayout::NCHW); ARMNN_AUTO_TEST_CASE(Convolution2dPerAxisQuantTestNhwc, Convolution2dPerAxisQuantTest, DataLayout::NHWC); +ARMNN_AUTO_TEST_CASE(Convolution2d3x3Stride2x2Bf16, Convolution2d3x3Stride2x2BFloat16Test, false, DataLayout::NHWC); +ARMNN_AUTO_TEST_CASE(Convolution2d3x3Stride2x2BFloat16SmallValue, + Convolution2d3x3Stride2x2BFloat16SmallValueTest, + false, + DataLayout::NHWC); + // Depthwise Convolution ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2d, DepthwiseConvolution2dTest, true, DataLayout::NCHW) ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dUint8, DepthwiseConvolution2dUint8Test, true, DataLayout::NCHW) |