diff options
| -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()) { |