commit bdeabf8abd6ac48b63b9cfdaeb76ea4dc6f2ce9c
author Ryan OShea <Ryan.OShea2@arm.com> Wed Mar 04 12:42:05 2020 +0000
committer Ryan O'Shea <ryan.oshea2@arm.com> Wed Mar 04 15:09:31 2020 +0000
tree f183df0aada70baf3b2d1b333b6d9e35dca77171
parent 8a2b4685fde869c46ad4ebb19cbfefc4adc2a654

IVGCVSW-4213 Enable split along any dimension

 * Remove check on split dimension

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

src/armnnTfLiteParser/TfLiteParser.cpp

diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp
index 56b59a1..5a5274a 100644
--- a/src/armnnTfLiteParser/TfLiteParser.cpp
+++ b/src/armnnTfLiteParser/TfLiteParser.cpp
@@ -2454,17 +2454,6 @@
     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)
     {