diff options
Diffstat (limited to 'src/armnn/layers/BatchNormalizationLayer.cpp')
| -rw-r--r-- | src/armnn/layers/BatchNormalizationLayer.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/layers/BatchNormalizationLayer.cpp b/src/armnn/layers/BatchNormalizationLayer.cpp index 2212f47d3c..725dbd88b2 100644 --- a/src/armnn/layers/BatchNormalizationLayer.cpp +++ b/src/armnn/layers/BatchNormalizationLayer.cpp @@ -75,7 +75,8 @@ void BatchNormalizationLayer::Accept(ILayerVisitor& visitor) const ConstTensor varianceTensor(m_Variance->GetTensorInfo(), m_Variance->Map(true)); ConstTensor betaTensor(m_Beta->GetTensorInfo(), m_Beta->Map(true)); ConstTensor gammaTensor(m_Gamma->GetTensorInfo(), m_Gamma->Map(true)); - visitor.VisitBatchNormalizationLayer(this, GetParameters(), meanTensor, varianceTensor, betaTensor, gammaTensor); + visitor.VisitBatchNormalizationLayer( + this, GetParameters(), meanTensor, varianceTensor, betaTensor, gammaTensor, GetName()); } } // namespace armnn |