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 69713145c0..f1b31571db 100644 --- a/src/backends/reference/workloads/RefWorkloadUtils.hpp +++ b/src/backends/reference/workloads/RefWorkloadUtils.hpp @@ -75,6 +75,12 @@ const BFloat16* GetInputTensorDataBFloat16(unsigned int idx, const PayloadType& return GetInputTensorData<BFloat16>(idx, data); } +template <typename PayloadType> +BFloat16* GetOutputTensorDataBFloat16(unsigned int idx, const PayloadType& data) +{ + return GetOutputTensorData<BFloat16>(idx, data); +} + //////////////////////////////////////////// /// u8 helpers //////////////////////////////////////////// |