aboutsummaryrefslogtreecommitdiff
path: root/src/armnnTfLiteParser/TfLiteParser.cpp
diff options
context:
space:
mode:
authorNikhil Raj Arm <nikhil.raj@arm.com>2023-08-11 08:55:21 +0000
committerNikhil Raj Arm <nikhil.raj@arm.com>2023-08-11 08:55:21 +0000
commit008270f8c1359a7d62c2f881326b4d3f0d8b7b56 (patch)
tree5496cc6b428e1ecc0740aaf62dba331c77a9a9eb /src/armnnTfLiteParser/TfLiteParser.cpp
parent4980e21193f0a14fef084a7f4b4197392f3c0845 (diff)
downloadarmnn-008270f8c1359a7d62c2f881326b4d3f0d8b7b56.tar.gz
Revert "MLCE-1093 Reshape and concat invalid results"
This reverts commit 4980e21193f0a14fef084a7f4b4197392f3c0845. Reason for revert: Android Build for v82a failed due to schema not re-generating Change-Id: Ic19cf471b487f321c97ff837d36526512fb12fa4
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-rw-r--r--src/armnnTfLiteParser/TfLiteParser.cpp8
1 files changed, 2 insertions, 6 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp
index f40f7ffe0e..301989ebb7 100644
--- a/src/armnnTfLiteParser/TfLiteParser.cpp
+++ b/src/armnnTfLiteParser/TfLiteParser.cpp
@@ -4213,7 +4213,7 @@ void TfLiteParserImpl::ParseUnpack(size_t subgraphIndex, size_t operatorIndex)
}
splitDesc.SetViewOriginCoord(j, unpackAxis, unpackDimSizes[unpackAxis] * j);
}
- splitDesc.SetAxis(unpackAxis);
+
auto layerName = fmt::format("Unpack:{}:{}", subgraphIndex, operatorIndex);
IConnectableLayer* layer = m_Network->AddSplitterLayer(splitDesc, layerName.c_str());
@@ -4352,10 +4352,7 @@ void TfLiteParserImpl::ParseSplit(size_t subgraphIndex, size_t operatorIndex)
}
splitDesc.SetViewOriginCoord(j, splitDim, splitterDimSizes[splitDim] * j);
}
- if (axisTensorInfo.GetNumElements() == 1)
- {
- splitDesc.SetAxis(axis);
- }
+
auto layerName = fmt::format("Split:{}:{}", subgraphIndex, operatorIndex);
IConnectableLayer* layer = m_Network->AddSplitterLayer(splitDesc, layerName.c_str());
@@ -4534,7 +4531,6 @@ void TfLiteParserImpl::ParseSplitV(size_t subgraphIndex, size_t operatorIndex)
splitDesc.SetViewOriginCoord(j, splitDim, accumSplit);
accumSplit += splitSize;
}
- splitDesc.SetAxis(axis);
auto layerName = fmt::format("SplitV:{}:{}", subgraphIndex, operatorIndex);
IConnectableLayer* layer = m_Network->AddSplitterLayer(splitDesc, layerName.c_str());