diff options
Diffstat (limited to 'src/armnnSerializer')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 4 | ||||
-rw-r--r-- | src/armnnSerializer/test/SerializerTests.cpp | 1 |
2 files changed, 1 insertions, 4 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 1b7e46e51a..6cadb598a2 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -1306,9 +1306,7 @@ void SerializerStrategy::SerializeSplitterLayer(const armnn::IConnectableLayer* // Create FlatBuffer ViewsDescriptor auto flatBufferViewsDescriptor = CreateViewsDescriptor(m_flatBufferBuilder, flatBufferOriginDescriptor, - m_flatBufferBuilder.CreateVector(flatBufferViewSizes), - viewsDescriptor.HasAxis(), - viewsDescriptor.GetAxis()); + m_flatBufferBuilder.CreateVector(flatBufferViewSizes)); // Create FlatBuffer BaseLayer auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Splitter); diff --git a/src/armnnSerializer/test/SerializerTests.cpp b/src/armnnSerializer/test/SerializerTests.cpp index bfe3fc6467..b2590eaa42 100644 --- a/src/armnnSerializer/test/SerializerTests.cpp +++ b/src/armnnSerializer/test/SerializerTests.cpp @@ -2644,7 +2644,6 @@ TEST_CASE("SerializeSplitter") desc.SetViewSize(g, dimIdx, splitterDimSizes[dimIdx]); } } - desc.SetAxis(1); const std::string layerName("splitter"); const armnn::TensorInfo inputInfo(numDimensions, inputShape, armnn::DataType::Float32); |