aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/test/ConvertFp16ToFp32TestImpl.hpp
diff options
context:
space:
mode:
authorAron Virginas-Tar <Aron.Virginas-Tar@arm.com>2018-11-14 18:35:18 +0000
committerAron Virginas-Tar <Aron.Virginas-Tar@arm.com>2018-11-15 10:38:19 +0000
commit5caf907efc31e774f8afde54b17a5596477772f6 (patch)
tree9fcdfe44ccf7c96e5088a2cef06b7d74dfd3221c /src/backends/backendsCommon/test/ConvertFp16ToFp32TestImpl.hpp
parentdd9d8ca997cb6c63677249350247e9f44525104c (diff)
downloadarmnn-5caf907efc31e774f8afde54b17a5596477772f6.tar.gz
IVGCVSW-2136: Remove memory management methods from workload factories
Change-Id: Idc0f94590566ac362f7e1d1999361d025cc2f67a
Diffstat (limited to 'src/backends/backendsCommon/test/ConvertFp16ToFp32TestImpl.hpp')
-rw-r--r--src/backends/backendsCommon/test/ConvertFp16ToFp32TestImpl.hpp7
1 files changed, 6 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/test/ConvertFp16ToFp32TestImpl.hpp b/src/backends/backendsCommon/test/ConvertFp16ToFp32TestImpl.hpp
index a63f0cbd1d..1a86f4cbac 100644
--- a/src/backends/backendsCommon/test/ConvertFp16ToFp32TestImpl.hpp
+++ b/src/backends/backendsCommon/test/ConvertFp16ToFp32TestImpl.hpp
@@ -5,6 +5,8 @@
#pragma once
+#include "WorkloadTestUtils.hpp"
+
#include <armnn/ArmNN.hpp>
#include <armnn/Tensor.hpp>
#include <armnn/TypesUtils.hpp>
@@ -12,10 +14,13 @@
#include <Half.hpp>
#include <backendsCommon/CpuTensorHandle.hpp>
+#include <backendsCommon/IBackendInternal.hpp>
#include <test/TensorHelpers.hpp>
-LayerTestResult<float, 4> SimpleConvertFp16ToFp32Test(armnn::IWorkloadFactory& workloadFactory)
+LayerTestResult<float, 4> SimpleConvertFp16ToFp32Test(
+ armnn::IWorkloadFactory& workloadFactory,
+ const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager)
{
using namespace half_float::literal;