From 01097941ef85073c56cbd1d5f00d7e8ffeb9876d Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Mon, 26 Apr 2021 12:06:34 +0100 Subject: IVGCVSW-5843 Separate memory managers for WorkingMemHandles * Add inter layer memory management to WorkingMemHandle * Change Const layers to be executed once in loadedNetworkConstruction and share tensorHandle between all WorkingMemHandles * Fix various reference workloads pointing to memory in the queueDescriptor Signed-off-by: Finn Williams Change-Id: I69d4b3c5c84d2f5abe4540c3e624ab4f00d88226 --- src/backends/reference/workloads/PreluImpl.cpp | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) (limited to 'src/backends/reference/workloads/PreluImpl.cpp') diff --git a/src/backends/reference/workloads/PreluImpl.cpp b/src/backends/reference/workloads/PreluImpl.cpp index 458025bb0a..6df259fa4d 100644 --- a/src/backends/reference/workloads/PreluImpl.cpp +++ b/src/backends/reference/workloads/PreluImpl.cpp @@ -10,15 +10,13 @@ namespace armnn { -void PreluImpl(const PreluQueueDescriptor& data, +void PreluImpl(const TensorInfo& inputInfo, + const TensorInfo& alphaInfo, + const TensorInfo& outputInfo, Decoder& inputData, Decoder& alphaData, Encoder& outputData) { - const TensorInfo& inputInfo = GetTensorInfo(data.m_Inputs[0]); - const TensorInfo& alphaInfo = GetTensorInfo(data.m_Inputs[1]); - const TensorInfo& outputInfo = GetTensorInfo(data.m_Outputs[0]); - const TensorShape& inputShape = inputInfo.GetShape(); const TensorShape& alphaShape = alphaInfo.GetShape(); const TensorShape& outputShape = outputInfo.GetShape(); -- cgit v1.2.1