diff options
Diffstat (limited to 'src/armnn/layers/GatherLayer.cpp')
-rw-r--r-- | src/armnn/layers/GatherLayer.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/layers/GatherLayer.cpp b/src/armnn/layers/GatherLayer.cpp index d7ed4b2542..e3ce6b3dad 100644 --- a/src/armnn/layers/GatherLayer.cpp +++ b/src/armnn/layers/GatherLayer.cpp @@ -60,4 +60,9 @@ void GatherLayer::ValidateTensorShapesFromInputs() inferredShape); } +void GatherLayer::Accept(ILayerVisitor& visitor) const +{ + visitor.VisitGatherLayer(this, GetName()); +} + } // namespace armnn |