diff options
Diffstat (limited to 'src/backends/reference/workloads/RefWorkloadUtils.hpp')
| -rw-r--r-- | src/backends/reference/workloads/RefWorkloadUtils.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/RefWorkloadUtils.hpp b/src/backends/reference/workloads/RefWorkloadUtils.hpp index c3260c8142..69713145c0 100644 --- a/src/backends/reference/workloads/RefWorkloadUtils.hpp +++ b/src/backends/reference/workloads/RefWorkloadUtils.hpp @@ -12,6 +12,7 @@ #include <reference/RefTensorHandle.hpp> +#include <BFloat16.hpp> #include <Half.hpp> #include <boost/polymorphic_cast.hpp> @@ -68,6 +69,12 @@ Half* GetOutputTensorDataHalf(unsigned int idx, const PayloadType& data) return GetOutputTensorData<Half>(idx, data); } +template <typename PayloadType> +const BFloat16* GetInputTensorDataBFloat16(unsigned int idx, const PayloadType& data) +{ + return GetInputTensorData<BFloat16>(idx, data); +} + //////////////////////////////////////////// /// u8 helpers //////////////////////////////////////////// |