From 611c7fb97412230d5cefee047081455fb60db06c Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Fri, 7 Jan 2022 09:47:29 +0000 Subject: IVGCVSW-6641 Stabilize the IWorkloadFactory interface with unified strategy Signed-off-by: Teresa Charlin Change-Id: Ia941be9bf2c15fe56e49a9b9a2bbe943a8152438 --- .../backendsCommon/test/layerTests/Pooling2dTestImpl.cpp | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) (limited to 'src/backends/backendsCommon/test/layerTests/Pooling2dTestImpl.cpp') diff --git a/src/backends/backendsCommon/test/layerTests/Pooling2dTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/Pooling2dTestImpl.cpp index 248f9726a1..643a5df6be 100644 --- a/src/backends/backendsCommon/test/layerTests/Pooling2dTestImpl.cpp +++ b/src/backends/backendsCommon/test/layerTests/Pooling2dTestImpl.cpp @@ -100,7 +100,9 @@ LayerTestResult SimplePooling2dTestImpl( return result; } - std::unique_ptr workload = workloadFactory.CreatePooling2d(queueDescriptor, workloadInfo); + std::unique_ptr workload = workloadFactory.CreateWorkload(armnn::LayerType::Pooling2d, + queueDescriptor, + workloadInfo); inputHandle->Allocate(); outputHandle->Allocate(); @@ -830,8 +832,10 @@ LayerTestResult ComparePooling2dTestCommon( SetWorkloadInput(refData, refInfo, 0, inputTensorInfo, inputHandleRef.get()); SetWorkloadOutput(refData, refInfo, 0, outputTensorInfo, outputHandleRef.get()); - std::unique_ptr workload = workloadFactory.CreatePooling2d(data, info); - std::unique_ptr workloadRef = refWorkloadFactory.CreatePooling2d(refData, refInfo); + std::unique_ptr workload + = workloadFactory.CreateWorkload(armnn::LayerType::Pooling2d, data, info); + std::unique_ptr workloadRef + = refWorkloadFactory.CreateWorkload(armnn::LayerType::Pooling2d, refData, refInfo); outputHandleRef->Allocate(); inputHandleRef->Allocate(); -- cgit v1.2.1