commit f0d8ec1f4be3ce5794a48c3d3449fe93f2e58d04
author Kevin May <kevin.may@arm.com> Thu Dec 14 14:57:59 2023 +0000
committer TeresaARM <teresa.charlinreyes@arm.com> Thu Dec 14 17:02:20 2023 +0000
tree 4af5e58e56688a841c19ef6b599f8834ccbca24a
parent 01665507036addb9f4cfc0d06b89edaca68116fd

Fix incorrect Split Tosa Ref outputs check

* Resolves IVGCVSW-7918

Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: Idd15ce139f55895957378f9a9d1471e3e48989bb

src/backends/tosaCommon/operatorMappings/SplitOperator.cpp

diff --git a/src/backends/tosaCommon/operatorMappings/SplitOperator.cpp b/src/backends/tosaCommon/operatorMappings/SplitOperator.cpp
index 5231f96..f8c60b1 100644
--- a/src/backends/tosaCommon/operatorMappings/SplitOperator.cpp
+++ b/src/backends/tosaCommon/operatorMappings/SplitOperator.cpp
@@ -18,8 +18,8 @@
     ARMNN_THROW_INVALIDARG_MSG_IF_FALSE( inputs.size() == 1,
                                          "ConvertSplitToTosaOperator: Split must have only one input" );
 
-    ARMNN_THROW_INVALIDARG_MSG_IF_FALSE( outputs.size() < 1,
-                                         "ConvertSplitToTosaOperator: Split must have more than one output" );
+    ARMNN_THROW_INVALIDARG_MSG_IF_FALSE( outputs.size() >= 1,
+                                         "ConvertSplitToTosaOperator: Split must have at least one output" );
 
     if (!inputs[0]->GetShape().AreAllDimensionsSpecified())
     {