aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.cpp
diff options
context:
space:
mode:
authorMike Kelly <mike.kelly@arm.com>2023-08-04 13:35:41 +0100
committermike.kelly <mike.kelly@arm.com>2023-08-10 15:45:09 +0000
commit4980e21193f0a14fef084a7f4b4197392f3c0845 (patch)
tree544abd9d77d5eb776114d380ddba854d583b10ca /src/armnnSerializer/Serializer.cpp
parent8259941977ce51254b32d5ff54bffc26a06c4780 (diff)
downloadarmnn-4980e21193f0a14fef084a7f4b4197392f3c0845.tar.gz
MLCE-1093 Reshape and concat invalid results
!android-nn-driver:10089 * Disabled SubTensors on CL and Neon Backends. * Added Axis to ViewsDescriptor to store the value where ever possible. * Updated Splitter tests to provide all the information needed in the Descriptor. * Updated Serializer and Deserializer to handle axis. Signed-off-by: Mike Kelly <mike.kelly@arm.com> Change-Id: I6a22d4750e04003689495b5e9e3c33deb37162bd
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r--src/armnnSerializer/Serializer.cpp4
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index 6cadb598a2..1b7e46e51a 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -1306,7 +1306,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);