diff options
author | Derek Lamberti <derek.lamberti@arm.com> | 2019-02-07 11:14:11 +0000 |
---|---|---|
committer | Derek Lamberti <derek.lamberti@arm.com> | 2019-02-07 13:21:28 +0000 |
commit | 857aa45407df9dbe99a11d03a4be2b20bd0110ae (patch) | |
tree | 3e47a2aa1ac8787a00900eff0ba49246ef9a4bdc /src/armnn/layers | |
parent | 49dbe0e9f6747583cff29ada68d6670796d4216c (diff) | |
download | armnn-857aa45407df9dbe99a11d03a4be2b20bd0110ae.tar.gz |
IVGCVSW-2609 Quantize BatchNormalizationLayer
Change-Id: I7b847112a0322ffc8b88a0708d8439bfb97cfe2c
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Diffstat (limited to 'src/armnn/layers')
-rw-r--r-- | src/armnn/layers/BatchNormalizationLayer.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
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<void*>()) ; - ConstTensor varianceTensor(m_Variance->GetTensorInfo(), m_Variance->GetTensor<void*>()) ; - ConstTensor betaTensor(m_Beta->GetTensorInfo(), m_Beta->GetTensor<void*>()) ; - ConstTensor gammaTensor(m_Gamma->GetTensorInfo(), m_Gamma->GetTensor<void*>()) ; + 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); } |