diff options
author | Mike Kelly <mike.kelly@arm.com> | 2023-10-11 14:25:50 +0100 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2023-10-12 11:57:55 +0000 |
commit | 363b572b61f7a32e92cde51478d7556ce43db56f (patch) | |
tree | c4f888c607acab8d8e683b77bb20f7c2e507c558 /src/armnnDeserializer | |
parent | c4a932a5f2f8b5f8b684ed2034768136493fc64b (diff) | |
download | armnn-363b572b61f7a32e92cde51478d7556ce43db56f.tar.gz |
Revert "Revert "MLCE-1093 Reshape and concat invalid results""
This reverts commit 008270f8c1359a7d62c2f881326b4d3f0d8b7b56.
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: If8f5151aa349ff3834f03391e813669e5c51ed66
Diffstat (limited to 'src/armnnDeserializer')
-rw-r--r-- | src/armnnDeserializer/Deserializer.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp index f455b1af0a..f27489f339 100644 --- a/src/armnnDeserializer/Deserializer.cpp +++ b/src/armnnDeserializer/Deserializer.cpp @@ -3263,6 +3263,11 @@ void IDeserializer::DeserializerImpl::ParseSplitter(GraphPtr graph, unsigned int } } + if (flatBufferViewsDescriptor->hasAxis()) + { + viewsDescriptor.SetAxis(flatBufferViewsDescriptor->axis()); + } + auto layerName = GetLayerName(graph, layerIndex); IConnectableLayer* layer = m_Network->AddSplitterLayer(viewsDescriptor, layerName.c_str()); |