diff options
| author | Kevin May <kevin.may@arm.com> | 2023-12-14 14:57:59 +0000 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2023-12-14 17:02:20 +0000 |
| commit | f0d8ec1f4be3ce5794a48c3d3449fe93f2e58d04 (patch) | |
| tree | 4af5e58e56688a841c19ef6b599f8834ccbca24a | |
| parent | 01665507036addb9f4cfc0d06b89edaca68116fd (diff) | |
| download | armnn-f0d8ec1f4be3ce5794a48c3d3449fe93f2e58d04.tar.gz | |
Fix incorrect Split Tosa Ref outputs check
* Resolves IVGCVSW-7918
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: Idd15ce139f55895957378f9a9d1471e3e48989bb
| -rw-r--r-- | src/backends/tosaCommon/operatorMappings/SplitOperator.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/tosaCommon/operatorMappings/SplitOperator.cpp b/src/backends/tosaCommon/operatorMappings/SplitOperator.cpp index 5231f96c36..f8c60b1b6d 100644 --- a/src/backends/tosaCommon/operatorMappings/SplitOperator.cpp +++ b/src/backends/tosaCommon/operatorMappings/SplitOperator.cpp @@ -18,8 +18,8 @@ TosaSerializationBasicBlock* ConvertSplitToTosaOperator(const Layer* layer, 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()) { |