From aebf2e7871997378d33fa99f7cab6322be9b726f Mon Sep 17 00:00:00 2001
From: jimfly01 <jim.flynn@arm.com>
Date: Tue, 5 Feb 2019 09:20:11 +0000
Subject: IVGCVSW-2553 Add Unit Test for ConstTensor layer visitors

Change-Id: I20a67135e4af7aa1f28f5000d73122f4e5e3acd5
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
---
 src/armnn/layers/DepthwiseConvolution2dLayer.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'src/armnn/layers/DepthwiseConvolution2dLayer.cpp')

diff --git a/src/armnn/layers/DepthwiseConvolution2dLayer.cpp b/src/armnn/layers/DepthwiseConvolution2dLayer.cpp
index dca13f263c..a17673fc1e 100644
--- a/src/armnn/layers/DepthwiseConvolution2dLayer.cpp
+++ b/src/armnn/layers/DepthwiseConvolution2dLayer.cpp
@@ -122,10 +122,10 @@ Layer::ConstantTensors DepthwiseConvolution2dLayer::GetConstantTensorsByRef()
 
 void DepthwiseConvolution2dLayer::Accept(ILayerVisitor& visitor) const
 {
-    ConstTensor weightsTensor(m_Weight->GetTensorInfo(), m_Weight->GetTensor<void*>()) ;
-    ConstTensor biasTensor(m_Bias->GetTensorInfo(), m_Bias->GetConstTensor<void*>());
+    ConstTensor weightsTensor(m_Weight->GetTensorInfo(), m_Weight->Map(true)) ;
     if (GetParameters().m_BiasEnabled)
     {
+        ConstTensor biasTensor(m_Bias->GetTensorInfo(), m_Bias->Map(true));
         visitor.VisitDepthwiseConvolution2dLayer(this, GetParameters(), weightsTensor, biasTensor, GetName());
     }
     else
-- 
cgit v1.2.1