diff options
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 // |