diff options
author | David Beck <david.beck@arm.com> | 2018-10-23 16:09:36 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-25 09:49:58 +0100 |
commit | 79141b9662547eeefb3ad533637223de40726e12 (patch) | |
tree | 9869465c1769fb1f6e2bf86306fbac3bc1b6add4 /src/backends/reference/RefWorkloadFactory.cpp | |
parent | 29c75de868ac3a86a70b25f8da0d0c7e47d40803 (diff) | |
download | armnn-79141b9662547eeefb3ad533637223de40726e12.tar.gz |
IVGCVSW-2071 : remove GetCompute() from the WorkloadFactory interface
Change-Id: I44a9d26d1a5d876d381aee4c6450af62811d0dbb
Diffstat (limited to 'src/backends/reference/RefWorkloadFactory.cpp')
-rw-r--r-- | src/backends/reference/RefWorkloadFactory.cpp | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/src/backends/reference/RefWorkloadFactory.cpp b/src/backends/reference/RefWorkloadFactory.cpp index 783e5fba2e..864ffdbf4f 100644 --- a/src/backends/reference/RefWorkloadFactory.cpp +++ b/src/backends/reference/RefWorkloadFactory.cpp @@ -6,6 +6,7 @@ #include <backends/MemCopyWorkload.hpp> #include <backends/MakeWorkloadHelper.hpp> #include "RefWorkloadFactory.hpp" +#include "RefBackendId.hpp" #include "workloads/RefWorkloads.hpp" #include "Layer.hpp" @@ -14,6 +15,11 @@ namespace armnn { +namespace +{ +static const BackendId s_Id{RefBackendId()}; +} + template <typename F32Workload, typename U8Workload, typename QueueDescriptorType> std::unique_ptr<IWorkload> RefWorkloadFactory::MakeWorkload(const QueueDescriptorType& descriptor, const WorkloadInfo& info) const @@ -25,11 +31,16 @@ RefWorkloadFactory::RefWorkloadFactory() { } +const BackendId& RefWorkloadFactory::GetBackendId() const +{ + return s_Id; +} + bool RefWorkloadFactory::IsLayerSupported(const Layer& layer, Optional<DataType> dataType, std::string& outReasonIfUnsupported) { - return IWorkloadFactory::IsLayerSupported(Compute::CpuRef, layer, dataType, outReasonIfUnsupported); + return IWorkloadFactory::IsLayerSupported(s_Id, layer, dataType, outReasonIfUnsupported); } std::unique_ptr<ITensorHandle> RefWorkloadFactory::CreateTensorHandle(const TensorInfo& tensorInfo) const |