From 9270d9e52fc1d946cf543c11df71bcc1f7b38544 Mon Sep 17 00:00:00 2001 From: Francis Murtagh Date: Fri, 12 Aug 2022 13:54:17 +0100 Subject: IVGCVSW-7159 Implement simple TOSA Reference Backend skeleton * Added files based on RefBackend * Added PreCompiled Workload skeleton * Increment ABI version of armnnTestUtils for CreateInput which had been left as pure virtual, added base implementation for it. * Add IsTosaLayerSupported() for Addition Change-Id: I4c963adf3f50593d17ecdf21554502a64ad3bd76 --- src/backends/reference/workloads/RefWorkloadUtils.hpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'src/backends/reference') diff --git a/src/backends/reference/workloads/RefWorkloadUtils.hpp b/src/backends/reference/workloads/RefWorkloadUtils.hpp index e50847fee5..7c35966f0f 100644 --- a/src/backends/reference/workloads/RefWorkloadUtils.hpp +++ b/src/backends/reference/workloads/RefWorkloadUtils.hpp @@ -23,11 +23,12 @@ namespace armnn /// float32 helpers //////////////////////////////////////////// +template inline const TensorInfo& GetTensorInfo(const ITensorHandle* tensorHandle) { // We know that reference workloads use RefTensorHandles for inputs and outputs - const RefTensorHandle* refTensorHandle = - PolymorphicDowncast(tensorHandle); + const TensorHandleType* refTensorHandle = + PolymorphicDowncast(tensorHandle); return refTensorHandle->GetTensorInfo(); } -- cgit v1.2.1