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