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/FullyConnectedLayer.cpp | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'src/armnn/layers/FullyConnectedLayer.cpp') diff --git a/src/armnn/layers/FullyConnectedLayer.cpp b/src/armnn/layers/FullyConnectedLayer.cpp index e003c7350a..977c2769df 100644 --- a/src/armnn/layers/FullyConnectedLayer.cpp +++ b/src/armnn/layers/FullyConnectedLayer.cpp @@ -86,4 +86,17 @@ Layer::ConstantTensors FullyConnectedLayer::GetConstantTensorsByRef() return {m_Weight, m_Bias}; } +void FullyConnectedLayer::Accept(ILayerVisitor& visitor) const +{ + ConstTensor dummy; + if (GetParameters().m_BiasEnabled) + { + visitor.VisitFullyConnectedLayer(this, GetParameters(), dummy, dummy, GetName()); + } + else + { + visitor.VisitFullyConnectedLayer(this, GetParameters(), dummy, GetName()); + } +} + } // namespace armnn -- cgit v1.2.1