diff options
Diffstat (limited to 'src/armnn/layers/Convolution2dLayer.cpp')
-rw-r--r-- | src/armnn/layers/Convolution2dLayer.cpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/src/armnn/layers/Convolution2dLayer.cpp b/src/armnn/layers/Convolution2dLayer.cpp index 4756e1daae..86eb71d61a 100644 --- a/src/armnn/layers/Convolution2dLayer.cpp +++ b/src/armnn/layers/Convolution2dLayer.cpp @@ -110,4 +110,17 @@ Layer::ConstantTensors Convolution2dLayer::GetConstantTensorsByRef() return {m_Weight, m_Bias}; } +void Convolution2dLayer::Accept(ILayerVisitor& visitor) const +{ + ConstTensor dummy; + if (GetParameters().m_BiasEnabled) + { + visitor.VisitConvolution2dLayer(this, GetParameters(), dummy, dummy, GetName()); + } + else + { + visitor.VisitConvolution2dLayer(this, GetParameters(), dummy, GetName()); + } +} + } // namespace armnn |