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