From 857aa45407df9dbe99a11d03a4be2b20bd0110ae Mon Sep 17 00:00:00 2001 From: Derek Lamberti Date: Thu, 7 Feb 2019 11:14:11 +0000 Subject: IVGCVSW-2609 Quantize BatchNormalizationLayer Change-Id: I7b847112a0322ffc8b88a0708d8439bfb97cfe2c Signed-off-by: Derek Lamberti --- src/armnn/layers/BatchNormalizationLayer.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src/armnn/layers') diff --git a/src/armnn/layers/BatchNormalizationLayer.cpp b/src/armnn/layers/BatchNormalizationLayer.cpp index 85132053fb..2212f47d3c 100644 --- a/src/armnn/layers/BatchNormalizationLayer.cpp +++ b/src/armnn/layers/BatchNormalizationLayer.cpp @@ -71,10 +71,10 @@ Layer::ConstantTensors BatchNormalizationLayer::GetConstantTensorsByRef() void BatchNormalizationLayer::Accept(ILayerVisitor& visitor) const { - ConstTensor meanTensor(m_Mean->GetTensorInfo(), m_Mean->GetTensor()) ; - ConstTensor varianceTensor(m_Variance->GetTensorInfo(), m_Variance->GetTensor()) ; - ConstTensor betaTensor(m_Beta->GetTensorInfo(), m_Beta->GetTensor()) ; - ConstTensor gammaTensor(m_Gamma->GetTensorInfo(), m_Gamma->GetTensor()) ; + ConstTensor meanTensor(m_Mean->GetTensorInfo(), m_Mean->Map(true)); + 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); } -- cgit v1.2.1