diff options
| author | Mike Kelly <mike.kelly@arm.com> | 2023-01-19 18:29:40 +0000 |
|---|---|---|
| committer | mike.kelly <mike.kelly@arm.com> | 2023-01-23 22:05:22 +0000 |
| commit | 2542a267dc3dfe2b9148b3944977a6864ef3c558 (patch) | |
| tree | 1548da9d99abec215e80c5e6b2e25181a092b8b3 /src/armnn/test/ShapeInferenceTests.cpp | |
| parent | c01da459e63461dc0ecfe855fd3254c938689386 (diff) | |
| download | armnn-2542a267dc3dfe2b9148b3944977a6864ef3c558.tar.gz | |
IVGCVSW-7453 Comparison does not Calculate its shape properly
* Fixed issue where ComparisonLayer wasn't calculating its output shape correctly.
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I37fe437b598bde694e519d6792182924bd0197cd
Diffstat (limited to 'src/armnn/test/ShapeInferenceTests.cpp')
| -rw-r--r-- | src/armnn/test/ShapeInferenceTests.cpp | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/src/armnn/test/ShapeInferenceTests.cpp b/src/armnn/test/ShapeInferenceTests.cpp index 333d12a3a2..7b5d73a4e5 100644 --- a/src/armnn/test/ShapeInferenceTests.cpp +++ b/src/armnn/test/ShapeInferenceTests.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2020 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2020-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -223,6 +223,26 @@ TEST_CASE("ComparisionTest") "comparision"); } +TEST_CASE("ComparisionTestSmallerRHS") +{ + ComparisonDescriptor descriptor; + descriptor.m_Operation = ComparisonOperation::Equal; + CreateGraphAndRunTest<ComparisonLayer>({{ 5, 7, 6, 2 }, { 1 }}, + {{ 5, 7, 6, 2 }}, + descriptor, + "comparision"); +} + +TEST_CASE("ComparisionTestSmallerLHS") +{ + ComparisonDescriptor descriptor; + descriptor.m_Operation = ComparisonOperation::Equal; + CreateGraphAndRunTest<ComparisonLayer>({{ 1 }, { 5, 7, 6, 2 }}, + {{ 5, 7, 6, 2 }}, + descriptor, + "comparision"); +} + TEST_CASE("ConcatTest") { ConcatDescriptor descriptor(2, 3); |