diff options
Diffstat (limited to 'src/armnn/layers/FullyConnectedLayer.cpp')
-rw-r--r-- | src/armnn/layers/FullyConnectedLayer.cpp | 13 |
1 files changed, 13 insertions, 0 deletions
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 |