diff options
Diffstat (limited to 'src/armnn/layers/BatchNormalizationLayer.cpp')
| -rw-r--r-- | src/armnn/layers/BatchNormalizationLayer.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnn/layers/BatchNormalizationLayer.cpp b/src/armnn/layers/BatchNormalizationLayer.cpp index f461c8061b..dfba2d7942 100644 --- a/src/armnn/layers/BatchNormalizationLayer.cpp +++ b/src/armnn/layers/BatchNormalizationLayer.cpp @@ -69,4 +69,10 @@ Layer::ConstantTensors BatchNormalizationLayer::GetConstantTensorsByRef() return {m_Mean, m_Variance, m_Beta, m_Gamma}; } +void BatchNormalizationLayer::Accept(ILayerVisitor& visitor) const +{ + ConstTensor dummy; + visitor.VisitBatchNormalizationLayer(this, GetParameters(), dummy, dummy, dummy, dummy); +} + } // namespace armnn |