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/armnnSerializer/Serializer.cpp | |
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/armnnSerializer/Serializer.cpp')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 0df675d1db..ef2ca48e04 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -1308,7 +1308,9 @@ void SerializerStrategy::SerializeSplitterLayer(const armnn::IConnectableLayer* // Create FlatBuffer ViewsDescriptor auto flatBufferViewsDescriptor = CreateViewsDescriptor(m_flatBufferBuilder, flatBufferOriginDescriptor, - m_flatBufferBuilder.CreateVector(flatBufferViewSizes)); + m_flatBufferBuilder.CreateVector(flatBufferViewSizes), + viewsDescriptor.HasAxis(), + viewsDescriptor.GetAxis()); // Create FlatBuffer BaseLayer auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Splitter); |