diff options
Diffstat (limited to 'src/backends/reference/workloads/RefWorkloadUtils.hpp')
| -rw-r--r-- | src/backends/reference/workloads/RefWorkloadUtils.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/RefWorkloadUtils.hpp b/src/backends/reference/workloads/RefWorkloadUtils.hpp index dfde58fdfe..0d839afc1c 100644 --- a/src/backends/reference/workloads/RefWorkloadUtils.hpp +++ b/src/backends/reference/workloads/RefWorkloadUtils.hpp @@ -45,6 +45,12 @@ DataType* GetOutputTensorData(unsigned int idx, const PayloadType& data) return reinterpret_cast<DataType*>(tensorHandle->Map()); } +template <typename DataType> +DataType* GetOutputTensorData(ITensorHandle* tensorHandle) +{ + return reinterpret_cast<DataType*>(tensorHandle->Map()); +} + template <typename PayloadType> const float* GetInputTensorDataFloat(unsigned int idx, const PayloadType& data) { |