diff options
author | Ryan OShea <Ryan.OShea2@arm.com> | 2020-03-04 12:42:05 +0000 |
---|---|---|
committer | Ryan O'Shea <ryan.oshea2@arm.com> | 2020-03-04 15:09:31 +0000 |
commit | bdeabf8abd6ac48b63b9cfdaeb76ea4dc6f2ce9c (patch) | |
tree | f183df0aada70baf3b2d1b333b6d9e35dca77171 /src | |
parent | 8a2b4685fde869c46ad4ebb19cbfefc4adc2a654 (diff) | |
download | armnn-bdeabf8abd6ac48b63b9cfdaeb76ea4dc6f2ce9c.tar.gz |
IVGCVSW-4213 Enable split along any dimension
* Remove check on split dimension
Signed-off-by: Ryan OShea <Ryan.OShea2@arm.com>
Change-Id: I03b88b8c8f1dcd842d9cc9d783ad7d0aa2c0ed3a
Diffstat (limited to 'src')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 11 |
1 files changed, 0 insertions, 11 deletions
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) { |