diff options
| author | James Conroy <james.conroy@arm.com> | 2019-10-08 15:41:34 +0100 |
|---|---|---|
| committer | Jim Flynn Arm <jim.flynn@arm.com> | 2019-10-10 08:48:39 +0000 |
| commit | c8724c7b9ff663538bd32ad789dbcc3e1aa88637 (patch) | |
| tree | 629a55e0f949fd9bede7e3b34cdc7ff337f9762c /src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp | |
| parent | 92bbcaed655d1dfe696f12f264599589b8ada602 (diff) | |
| download | armnn-c8724c7b9ff663538bd32ad789dbcc3e1aa88637.tar.gz | |
IVGCVSW-3944 Add ArgMinMax output shape validation
Signed-off-by: James Conroy <james.conroy@arm.com>
Change-Id: I469895da158b062cd19248832525fa21527f7d41
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp')
| -rw-r--r-- | src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp index e023d60bf0..be7ef4e32e 100644 --- a/src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp +++ b/src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp @@ -190,7 +190,7 @@ LayerTestResult<int32_t, 3> ArgMaxHeightTest( const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) { const armnn::TensorShape inputShape{ 1, 3, 2, 4}; - const armnn::TensorShape outputShape{ 3, 1, 4 }; + const armnn::TensorShape outputShape{ 1, 3, 4 }; armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); armnn::TensorInfo outputTensorInfo(outputShape, armnn::DataType::Signed32); @@ -219,7 +219,7 @@ LayerTestResult<int32_t, 3> ArgMinWidthTest( const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) { const armnn::TensorShape inputShape{ 1, 3, 2, 4}; - const armnn::TensorShape outputShape{ 3, 2, 1 }; + const armnn::TensorShape outputShape{ 1, 3, 2 }; armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); armnn::TensorInfo outputTensorInfo(outputShape, armnn::DataType::Signed32); |