aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/ActivationLayer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/layers/ActivationLayer.cpp')
-rw-r--r--src/armnn/layers/ActivationLayer.cpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/layers/ActivationLayer.cpp b/src/armnn/layers/ActivationLayer.cpp
index 401ab299c3..6f80cce968 100644
--- a/src/armnn/layers/ActivationLayer.cpp
+++ b/src/armnn/layers/ActivationLayer.cpp
@@ -42,4 +42,9 @@ void ActivationLayer::ValidateTensorShapesFromInputs()
inferredShapes[0]);
}
+void ActivationLayer::Accept(ILayerVisitor& visitor) const
+{
+ visitor.VisitActivationLayer(this, GetParameters(), GetName());
+}
+
} // namespace armnn