diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-05-04 17:54:19 +0100 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-05-04 17:54:19 +0100 |
| commit | 38b72e8de898d84a1481e242803da61009719891 (patch) | |
| tree | e6a536674e53557c94862fd88fe365735e96d55b /src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp | |
| parent | f0fce5be82fa35ee816945b557561854e7bde7c0 (diff) | |
| download | armnn-38b72e8de898d84a1481e242803da61009719891.tar.gz | |
IVGCVSW-6937 Add INT32 support to FLOOR
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I6f7cddb2d23c67ae682132d18f98776c074dcb3b
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp')
| -rw-r--r-- | src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp index 2bf8c97af5..377c2aaf1a 100644 --- a/src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp +++ b/src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp @@ -19,10 +19,10 @@ LayerTestResult<T, 4> SimpleFloorTest( { IgnoreUnused(memoryManager); armnn::TensorInfo inputTensorInfo({1, 3, 2, 3}, ArmnnType); - inputTensorInfo.SetQuantizationScale(0.1f); + inputTensorInfo.SetQuantizationScale(1.0f); armnn::TensorInfo outputTensorInfo(inputTensorInfo); - outputTensorInfo.SetQuantizationScale(0.1f); + outputTensorInfo.SetQuantizationScale(1.0f); std::vector<T> input = ConvertToDataType<ArmnnType>( { @@ -80,9 +80,14 @@ SimpleFloorTest<armnn::DataType::Float16>( const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, const armnn::ITensorHandleFactory& tensorHandleFactory); - template LayerTestResult<armnn::ResolveType<armnn::DataType::QSymmS16>, 4> SimpleFloorTest<armnn::DataType::QSymmS16>( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, const armnn::ITensorHandleFactory& tensorHandleFactory); + +template LayerTestResult<armnn::ResolveType<armnn::DataType::Signed32>, 4> +SimpleFloorTest<armnn::DataType::Signed32>( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory);
\ No newline at end of file |