From bdeabf8abd6ac48b63b9cfdaeb76ea4dc6f2ce9c Mon Sep 17 00:00:00 2001 From: Ryan OShea Date: Wed, 4 Mar 2020 12:42:05 +0000 Subject: IVGCVSW-4213 Enable split along any dimension * Remove check on split dimension Signed-off-by: Ryan OShea Change-Id: I03b88b8c8f1dcd842d9cc9d783ad7d0aa2c0ed3a --- src/armnnTfLiteParser/TfLiteParser.cpp | 11 ----------- 1 file changed, 11 deletions(-) (limited to 'src') diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 56b59a115f..5a5274a90f 100644 --- 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) { -- cgit v1.2.1