diff options
Diffstat (limited to 'src/armnnDeserializer/Deserializer.cpp')
-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()); |