commit c81855f6ca52eb025a303b95eee7a12a5e2f9557
author Derek Lamberti <derek.lamberti@arm.com> Thu Jun 13 17:34:19 2019 +0100
committer Matteo Martincigh <matteo.martincigh@arm.com> Fri Jun 14 16:41:17 2019 +0000
tree 86ac342839f49348b54f29c4588b6a9e4059cd84
parent db48288f54ad441f34108cf270ed3089d3a465ea

IVGCVSW-3278 Cl and Neon TensorHandles inherit from common base interface

Change-Id: Ia68da09d8f0fb0a04af9cb61062d7edaa5f1b887
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>

src/backends/neon/workloads/NeonConvolution2dWorkload.cpp

diff --git a/src/backends/neon/workloads/NeonConvolution2dWorkload.cpp b/src/backends/neon/workloads/NeonConvolution2dWorkload.cpp
index 1080f32..a8137a2 100644
--- a/src/backends/neon/workloads/NeonConvolution2dWorkload.cpp
+++ b/src/backends/neon/workloads/NeonConvolution2dWorkload.cpp
@@ -60,8 +60,8 @@
 
     // todo: check tensor shapes match.
 
-    arm_compute::ITensor& input = boost::polymorphic_downcast<INeonTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
-    arm_compute::ITensor& output = boost::polymorphic_downcast<INeonTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
+    arm_compute::ITensor& input = boost::polymorphic_downcast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
+    arm_compute::ITensor& output = boost::polymorphic_downcast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
 
     arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout);
     input.info()->set_data_layout(aclDataLayout);