diff options
Diffstat (limited to 'src/backends/backendsCommon/test')
-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); |