diff options
Diffstat (limited to 'src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp')
-rw-r--r-- | src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp b/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp index 30972693de..2b3a314888 100644 --- a/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp +++ b/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp @@ -7,6 +7,9 @@ #include "armnn/backends/Workload.hpp" +#include <graph_status.h> +#include <model_runner.h> + #include <memory> #include <string> #include <vector> @@ -38,6 +41,14 @@ private: { this->m_Data.m_Outputs[slot] = tensorHandle; } + + template <typename T> + void SetInput(TosaReference::IModelRunner& runner, std::string inputName, uint32_t inputIndex) const; + + template <typename T> + void GetOutput(TosaReference::IModelRunner& runner, std::string outputName, uint32_t outputIndex) const; + + WorkloadInfo m_workloadInfo; }; } //namespace armnn |