diff options
Diffstat (limited to 'src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp')
| -rw-r--r-- | src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp b/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp index 2b3a314888..337e8f9572 100644 --- a/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp +++ b/src/backends/tosaReference/workloads/TosaRefPreCompiledWorkload.hpp @@ -42,9 +42,15 @@ private: this->m_Data.m_Outputs[slot] = tensorHandle; } + template <typename T, typename Trunner> + void SetInput(TosaReference::IModelRunner& runner, std::string inputName, uint32_t inputIndex) const; + template <typename T> void SetInput(TosaReference::IModelRunner& runner, std::string inputName, uint32_t inputIndex) const; + template <typename T, typename Trunner> + void GetOutput(TosaReference::IModelRunner& runner, std::string outputName, uint32_t outputIndex) const; + template <typename T> void GetOutput(TosaReference::IModelRunner& runner, std::string outputName, uint32_t outputIndex) const; |