aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnSerializer')
-rw-r--r--src/armnnSerializer/Serializer.cpp4
-rw-r--r--src/armnnSerializer/test/SerializerTests.cpp1
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);