aboutsummaryrefslogtreecommitdiff
path: root/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp
diff options
context:
space:
mode:
authorFrancis Murtagh <francis.murtagh@arm.com>2022-08-12 13:54:17 +0100
committerNikhil Raj <nikhil.raj@arm.com>2022-09-07 14:08:58 +0100
commit9270d9e52fc1d946cf543c11df71bcc1f7b38544 (patch)
tree4b763ed4d200f76e170462a1b84a476d1c73a648 /src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp
parentce8204a528b3973b2dbf258095ab048a061263ea (diff)
downloadarmnn-9270d9e52fc1d946cf543c11df71bcc1f7b38544.tar.gz
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
Diffstat (limited to 'src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp')
-rw-r--r--src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp43
1 files changed, 43 insertions, 0 deletions
diff --git a/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp b/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp
new file mode 100644
index 0000000000..30972693de
--- /dev/null
+++ b/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp
@@ -0,0 +1,43 @@
+//
+// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include "armnn/backends/Workload.hpp"
+
+#include <memory>
+#include <string>
+#include <vector>
+
+namespace armnn
+{
+
+bool TosaRefPreCompiledWorkloadValidate(std::string* reasonIfUnsupported);
+
+class TosaRefPreCompiledWorkload : public BaseWorkload<PreCompiledQueueDescriptor>
+{
+public:
+ TosaRefPreCompiledWorkload(const PreCompiledQueueDescriptor& descriptor,
+ const WorkloadInfo& info);
+ void Execute() const override;
+
+private:
+ bool SupportsTensorHandleReplacement() const override
+ {
+ return true;
+ }
+
+ void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override
+ {
+ this->m_Data.m_Inputs[slot] = tensorHandle;
+ }
+
+ void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override
+ {
+ this->m_Data.m_Outputs[slot] = tensorHandle;
+ }
+};
+
+} //namespace armnn