aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/SplitterLayer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/layers/SplitterLayer.cpp')
-rw-r--r--src/armnn/layers/SplitterLayer.cpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/layers/SplitterLayer.cpp b/src/armnn/layers/SplitterLayer.cpp
index 5cf1f6b3b1..b3a1094118 100644
--- a/src/armnn/layers/SplitterLayer.cpp
+++ b/src/armnn/layers/SplitterLayer.cpp
@@ -98,4 +98,9 @@ void SplitterLayer::ValidateTensorShapesFromInputs()
}
}
+void SplitterLayer::Accept(ILayerVisitor& visitor) const
+{
+ visitor.VisitSplitterLayer(this, GetParameters(), GetName());
+}
+
} // namespace armnn