From e9e7bfd09e3435d2bbb334e07c6a0a2514c80048 Mon Sep 17 00:00:00 2001 From: jimfly01 Date: Thu, 24 Jan 2019 22:29:33 +0000 Subject: IVGCVSW-2547 Add Accept function to IConnectableLayer * Layers which callback VisitXXXLayer with ConstTensor have been implemented with dummies to speed up implementation and unblock IVGCVSW-2531 Change-Id: I49b8035f12ec72d6bd6cee95075692f98c48e193 --- src/armnn/layers/SoftmaxLayer.cpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/armnn/layers/SoftmaxLayer.cpp') 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 -- cgit v1.2.1