IVGCVSW-4213 Enable split along any dimension

 * Remove check on split dimension

Signed-off-by: Ryan OShea <Ryan.OShea2@arm.com>
Change-Id: I03b88b8c8f1dcd842d9cc9d783ad7d0aa2c0ed3a

diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp
index 56b59a1..5a5274a 100644 (file)
--- a/src/armnnTfLiteParser/TfLiteParser.cpp
+++ b/src/armnnTfLiteParser/TfLiteParser.cpp
@@ -2454,17 +2454,6 @@ void TfLiteParser::ParseSplit(size_t subgraphIndex, size_t operatorIndex)
     BOOST_ASSERT(axisTensorInfo.GetNumElements() == 1);
     const unsigned int splitDim = axisData[0];
 
-    // Armnn supports split along the channel dimension for data formats NHWC and NCHW.
-    if (splitDim == 0 || splitDim == 2)
-    {
-        throw ParseException(
-            boost::str(
-                boost::format(
-                    "Dimension %1% for split is not supported by Armnn. %2%")
-                    % splitDim
-                    % CHECK_LOCATION().AsString()));
-    }
-
     auto inputDimSize = inputTensorInfo.GetNumDimensions();
     if (inputDimSize > MaxNumOfTensorDimensions)
     {