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/ConstantLayer.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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

diff --git a/src/armnn/layers/ConstantLayer.cpp b/src/armnn/layers/ConstantLayer.cpp
index 919fd61087..31e1549e0e 100644
--- a/src/armnn/layers/ConstantLayer.cpp
+++ b/src/armnn/layers/ConstantLayer.cpp
@@ -53,7 +53,7 @@ void ConstantLayer::ValidateTensorShapesFromInputs()
 
 void ConstantLayer::Accept(ILayerVisitor& visitor) const
 {
-    ConstTensor layerOutputTensor(m_LayerOutput->GetTensorInfo(), m_LayerOutput->GetTensor<void*>()) ;
+    ConstTensor layerOutputTensor(m_LayerOutput->GetTensorInfo(), m_LayerOutput->Map(true)) ;
     visitor.VisitConstantLayer(this, layerOutputTensor, GetName());
 }
 
-- 
cgit v1.2.1