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