From 4980e21193f0a14fef084a7f4b4197392f3c0845 Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Fri, 4 Aug 2023 13:35:41 +0100 Subject: 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 Change-Id: I6a22d4750e04003689495b5e9e3c33deb37162bd --- src/armnnDeserializer/Deserializer.cpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/armnnDeserializer') diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp index eb77f92842..8ca1e83dfe 100644 --- a/src/armnnDeserializer/Deserializer.cpp +++ b/src/armnnDeserializer/Deserializer.cpp @@ -3228,6 +3228,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()); -- cgit v1.2.1