From 363b572b61f7a32e92cde51478d7556ce43db56f Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Wed, 11 Oct 2023 14:25:50 +0100 Subject: Revert "Revert "MLCE-1093 Reshape and concat invalid results"" This reverts commit 008270f8c1359a7d62c2f881326b4d3f0d8b7b56. Signed-off-by: Mike Kelly Change-Id: If8f5151aa349ff3834f03391e813669e5c51ed66 --- delegate/classic/src/Split.hpp | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'delegate/classic/src/Split.hpp') diff --git a/delegate/classic/src/Split.hpp b/delegate/classic/src/Split.hpp index aaa610259f..57b7f8074e 100644 --- a/delegate/classic/src/Split.hpp +++ b/delegate/classic/src/Split.hpp @@ -107,6 +107,8 @@ TfLiteStatus VisitSplitOperator(DelegateData& delegateData, splitterDimSizes[splitDim] /= numSplits; armnn::SplitterDescriptor splitDescriptor(numSplits, inputDimSize); + splitDescriptor.SetAxis(axis); + for (unsigned int j = 0; j < numSplits; ++j) { // Set the size of the views. @@ -301,6 +303,8 @@ TfLiteStatus VisitSplitVOperator(DelegateData& delegateData, } armnn::SplitterDescriptor splitDescriptor(numSplits, inputDimSize); + splitDescriptor.SetAxis(axis); + unsigned int accumSplit = 0; for (unsigned int j = 0; j < numSplits; ++j) { -- cgit v1.2.1