aboutsummaryrefslogtreecommitdiff
path: root/src/armnnDeserializer/Deserializer.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/armnnDeserializer/Deserializer.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/armnnDeserializer/Deserializer.cpp')
-rw-r--r--src/armnnDeserializer/Deserializer.cpp5
1 files changed, 5 insertions, 0 deletions
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());