aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/Decoders.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/Decoders.hpp')
-rw-r--r--src/backends/reference/workloads/Decoders.hpp4
1 files changed, 4 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/Decoders.hpp b/src/backends/reference/workloads/Decoders.hpp
index 6a8c756048..83c57c1169 100644
--- a/src/backends/reference/workloads/Decoders.hpp
+++ b/src/backends/reference/workloads/Decoders.hpp
@@ -102,6 +102,10 @@ inline std::unique_ptr<Decoder<float>> MakeDecoder(const TensorInfo& info, const
info.GetQuantizationScale(),
info.GetQuantizationOffset());
}
+ case DataType::BFloat16:
+ {
+ return std::make_unique<BFloat16Decoder>(static_cast<const BFloat16*>(data));
+ }
case DataType::Float16:
{
return std::make_unique<Float16Decoder>(static_cast<const Half*>(data));