diff options
Diffstat (limited to 'src/armnn/layers/SoftmaxLayer.cpp')
| -rw-r--r-- | src/armnn/layers/SoftmaxLayer.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/layers/SoftmaxLayer.cpp b/src/armnn/layers/SoftmaxLayer.cpp index 4c2456d49c..90891b417a 100644 --- a/src/armnn/layers/SoftmaxLayer.cpp +++ b/src/armnn/layers/SoftmaxLayer.cpp @@ -43,4 +43,9 @@ void SoftmaxLayer::ValidateTensorShapesFromInputs() inferredShapes[0]); } +void SoftmaxLayer::Accept(ILayerVisitor& visitor) const +{ + visitor.VisitSoftmaxLayer(this, GetParameters(), GetName()); +} + } // namespace armnn |