diff options
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); |